Home
last modified time | relevance | path

Searched refs:BatchNormTraining (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dbfloat16_test.cc81 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()
Dbatch_normalization_test.cc243 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/
Dbatchnorm_expander_test.cc56 TEST_F(BatchNormExpanderTest, BatchNormTraining) { in TEST_F() argument
Dhlo_parser_test.cc756 "BatchNormTraining", in CreateTestCases()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc94 xla::XlaOp output = xla::BatchNormTraining( in CompileImpl()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment_test.cc182 TEST_F(LayoutAssignmentTest, BatchNormTraining) { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h884 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,
Dxla_builder.cc2641 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/
Doperation_semantics.md212 : : : `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>