Searched refs:BatchNormInference (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 90 xla::XlaOp output = xla::BatchNormInference( in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 109 TEST_F(LayoutAssignmentTest, BatchNormInference) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 573 XlaOp BatchNormInference(const XlaOp& operand, const XlaOp& scale,
981 friend XlaOp BatchNormInference(const XlaOp& operand, const XlaOp& scale,
1896 XlaOp BatchNormInference(const XlaOp& operand, const XlaOp& scale,
|
| D | xla_builder.cc | 2068 XlaOp XlaBuilder::BatchNormInference(const XlaOp& operand, const XlaOp& scale, in BatchNormInference() function in xla::XlaBuilder
3493 XlaOp BatchNormInference(const XlaOp& operand, const XlaOp& scale, in BatchNormInference() function
3497 return operand.builder()->BatchNormInference( in BatchNormInference()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_parser_test.cc | 669 "BatchNormInference", in CreateTestCases()
672 ENTRY %BatchNormInference.v6 (input: f32[2,2,2,2], offset: f32[2], scale: f32[2], mean: f32[2], var… in CreateTestCases()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 642 BatchNormInference(input_activations, scale_activations, offset_activations, in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 169 ## BatchNormInference section in Operation Semantics
172 [`XlaBuilder::BatchNormInference`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/x…
178 <b> `BatchNormInference(operand, scale, offset, mean, variance, epsilon, feature_index)` </b>
196 `BatchNormInference` is equivalent to calling `BatchNormTraining` without
199 latency in inference, hence the name `BatchNormInference`.
|