/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | bfloat16_test.cc | 81 XLA_TEST_F(Bfloat16Test, DISABLED_ON_INTERPRETER(BatchNormTraining)) { in XLA_TEST_F() argument 98 BatchNormTraining(operand, scale, offset, /*epsilon=*/0.001, kFeatureIndex); in XLA_TEST_F()
|
D | batch_normalization_test.cc | 243 BatchNormTraining(operand, scale, offset, in XLA_TEST_P() 268 auto output = BatchNormTraining(input_f32, scale, offset, in XLA_TEST_P() 297 BatchNormTraining(operand, scale, offset, in XLA_TEST_P() 322 auto output = BatchNormTraining(input_f32, scale, offset, in XLA_TEST_P() 357 BatchNormTraining(h0, h1, h2, in XLA_TEST_P() 389 BatchNormTraining(h0, h1, h2, in XLA_TEST_P() 635 BatchNormTraining(input_activations, scale_activations, offset_activations, in XLA_TEST_P()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | batchnorm_expander_test.cc | 56 TEST_F(BatchNormExpanderTest, BatchNormTraining) { in TEST_F() argument
|
D | hlo_parser_test.cc | 756 "BatchNormTraining", in CreateTestCases()
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | batch_norm_op.cc | 94 xla::XlaOp output = xla::BatchNormTraining( in CompileImpl()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_layout_assignment_test.cc | 182 TEST_F(LayoutAssignmentTest, BatchNormTraining) { in TEST_F() argument
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.h | 884 XlaOp BatchNormTraining(XlaOp operand, XlaOp scale, XlaOp offset, 1403 friend XlaOp BatchNormTraining(XlaOp operand, XlaOp scale, XlaOp offset, 2509 XlaOp BatchNormTraining(XlaOp operand, XlaOp scale, XlaOp offset, float epsilon,
|
D | xla_builder.cc | 2641 XlaOp XlaBuilder::BatchNormTraining(XlaOp operand, XlaOp scale, XlaOp offset, in BatchNormTraining() function in xla::XlaBuilder 4939 XlaOp BatchNormTraining(const XlaOp operand, const XlaOp scale, in BatchNormTraining() function 4942 return operand.builder()->BatchNormTraining(operand, scale, offset, epsilon, in BatchNormTraining()
|
/external/tensorflow/tensorflow/compiler/xla/g3doc/ |
D | operation_semantics.md | 212 : : : `BatchNormTraining` : 282 `BatchNormInference` is equivalent to calling `BatchNormTraining` without 290 ## BatchNormTraining section in Operation Semantics 293 [`XlaBuilder::BatchNormTraining`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xl… 299 <b> `BatchNormTraining(operand, scale, offset, epsilon, feature_index)` </b>
|