Searched refs:input_activations (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 138 auto input_activations = ConstantLiteral(&builder, input_literal_); in XLA_TEST_P() local
141 Reduce(input_activations, ConstantR0<float>(&builder, 0.0f), add, {0, 2, 3}); in XLA_TEST_P()
149 auto input_activations = ConstantLiteral(&builder, input_literal_); in XLA_TEST_P() local
151 auto activation_deviations = Sub(input_activations, set_means, in XLA_TEST_P()
174 auto input_activations = in XLA_TEST_P() local
184 Reduce(input_activations, ConstantR0<float>(&builder, 0.0f), add, in XLA_TEST_P()
187 auto input_shape = builder.GetShape(input_activations).ConsumeValueOrDie(); in XLA_TEST_P()
197 auto activation_deviations = Sub(input_activations, set_means, in XLA_TEST_P()
519 auto input_activations = in XLA_TEST_P() local
537 BatchNormTraining(input_activations, scale_activations, offset_activations, in XLA_TEST_P()
[all …]
|
| D | reduce_test.cc | 824 auto input_activations = in XLA_TEST_P() local
827 Reduce(input_activations, ConstantR0<float>(&builder, 0.0f), add, in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | hardcode_min_max.cc | 338 auto& input_activations = in HardcodeMinMaxForLstmCell() local
340 if (!input_activations.minmax) { in HardcodeMinMaxForLstmCell()
341 auto& minmax = input_activations.GetOrCreateMinMax(); in HardcodeMinMaxForLstmCell()
|
| D | quantize.cc | 217 const auto& input_activations = in ChooseQuantizationForOperatorInput() local
220 if (!input_activations.quantization_params || in ChooseQuantizationForOperatorInput()
227 input_activations.quantization_params->scale; in ChooseQuantizationForOperatorInput()
|