Home
last modified time | relevance | path

Searched refs:input_activations (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dbatch_normalization_test.cc138 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 …]
Dreduce_test.cc824 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/
Dhardcode_min_max.cc338 auto& input_activations = in HardcodeMinMaxForLstmCell() local
340 if (!input_activations.minmax) { in HardcodeMinMaxForLstmCell()
341 auto& minmax = input_activations.GetOrCreateMinMax(); in HardcodeMinMaxForLstmCell()
Dquantize.cc217 const auto& input_activations = in ChooseQuantizationForOperatorInput() local
220 if (!input_activations.quantization_params || in ChooseQuantizationForOperatorInput()
227 input_activations.quantization_params->scale; in ChooseQuantizationForOperatorInput()