| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_norm_op_gpu.cu.cc | 29 template struct functor::BatchNormGrad<GPUDevice, float>; member in tensorflow::functor
30 template struct functor::BatchNormGrad<GPUDevice, Eigen::half>; member in tensorflow::functor
|
| D | batch_norm_op.cc | 154 functor::BatchNormGrad<Device, T>()( in Compute()
239 void BatchNormGrad<GPUDevice, T>::operator()( \
248 extern template struct BatchNormGrad<GPUDevice, T>;
|
| D | batch_norm_op.h | 74 struct BatchNormGrad { struct
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | bfloat16_test.cc | 117 XLA_TEST_F(Bfloat16Test, DISABLED_ON_INTERPRETER(BatchNormGrad)) { in XLA_TEST_F() argument
140 BatchNormGrad(operand, scale, mean, var, grad_output, in XLA_TEST_F()
|
| D | batch_normalization_test.cc | 359 BatchNormGrad(operand, scale, mean, var, grad_output, in XLA_TEST_P()
827 BatchNormGrad(input_parameter, scale_parameter, mean_parameter, var_parameter, in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander_test.cc | 91 TEST_F(BatchNormExpanderTest, BatchNormGrad) { in TEST_F() argument
|
| D | hlo_element_type_converter_test.cc | 112 op::BatchNormGrad(); in TEST_F()
|
| D | hlo_matchers.h | 186 HLO_MATCHER(BatchNormGrad);
|
| D | hlo_parser_test.cc | 685 "BatchNormGrad", in CreateTestCases()
688 ENTRY %BatchNormGrad.v4 (input: f32[2,2,2,2], scale: f32[2], mean: f32[2], variance: f32[2], grad_o… in CreateTestCases()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 166 xla::BatchNormGrad(activations, scale, mean, var, grad_backprop, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 252 TEST_F(LayoutAssignmentTest, BatchNormGrad) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 578 XlaOp BatchNormGrad(const XlaOp& operand, const XlaOp& scale,
985 friend XlaOp BatchNormGrad(const XlaOp& operand, const XlaOp& scale,
1910 XlaOp BatchNormGrad(const XlaOp& operand, const XlaOp& scale,
|
| D | xla_builder.cc | 2094 XlaOp XlaBuilder::BatchNormGrad(const XlaOp& operand, const XlaOp& scale, in BatchNormGrad() function in xla::XlaBuilder
3501 XlaOp BatchNormGrad(const XlaOp& operand, const XlaOp& scale, in BatchNormGrad() function
3505 return operand.builder()->BatchNormGrad(operand, scale, batch_mean, batch_var, in BatchNormGrad()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 105 ## BatchNormGrad section in Operation Semantics
108 [`XlaBuilder::BatchNormGrad`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_bu…
114 <b> `BatchNormGrad(operand, scale, mean, variance, grad_output, epsilon, feature_index)` </b>
|