OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2026-01-15 12:15:41 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Previously we didn't have an implementation of BatchNormInference and BatchNormTraining, which gives a linker error if anyone ever tries to call that. A dummy implementation is friendlier than a linker error.

Browse Source 
PiperOrigin-RevId: 159645612
This commit is contained in:
A. Unique TensorFlower
2017-06-20 18:45:55 -07:00
committed by TensorFlower Gardener
parent f671c5caad
commit 26239c706f
1 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tensorflow/compiler/xla/client/computation_builder.cc
View File

@@ -1438,6 +1438,26 @@ ComputationDataHandle ComputationBuilder::BatchNormTraining(
return ParseOpResponse(s, &response);
}
ComputationDataHandle ComputationBuilder::BatchNormInference(
const ComputationDataHandle& operand, const ComputationDataHandle& scale,
const ComputationDataHandle& offset, const ComputationDataHandle& mean,
const ComputationDataHandle& variance, float epsilon, int64 feature_index) {
// TODO(b/62843645): Implement BatchNormInference.
NoteError(Unimplemented("BatchNormInference is not implemented yet."));
return ComputationDataHandle();
}
ComputationDataHandle ComputationBuilder::BatchNormGrad(
const ComputationDataHandle& operand, const ComputationDataHandle& scale,
const ComputationDataHandle& batch_mean,
const ComputationDataHandle& batch_var,
const ComputationDataHandle& grad_output, float epsilon,
int64 feature_index) {
// TODO(b/62843645): Implement BatchNormGrad.
NoteError(Unimplemented("BatchNormGrad is not implemented yet."));
return ComputationDataHandle();
}
ComputationDataHandle ComputationBuilder::CrossReplicaSum(
const ComputationDataHandle& operand) {
if (!first_error_.ok() || !PrepareComputation().ok()) {