Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_binary.cc72 const auto& input1_array = model->GetArray(binary_op->inputs[1]); in EvaluateBinaryOperatorOnConstantInputs() local
76 CHECK(input1_array.data_type == InputsDataType); in EvaluateBinaryOperatorOnConstantInputs()
82 CHECK(input1_array.buffer); in EvaluateBinaryOperatorOnConstantInputs()
87 const auto& input1_data = input1_array.GetBuffer<InputsDataType>().data; in EvaluateBinaryOperatorOnConstantInputs()
100 Shape input1_shape = input1_array.shape(); in EvaluateBinaryOperatorOnConstantInputs()
215 const auto& input1_array = model->GetArray(binary_op->inputs[1]); in Run() local
217 if (!input0_array.buffer || !input1_array.buffer) { in Run()
239 CHECK(input0_array.data_type == input1_array.data_type) in Run()
244 << static_cast<int>(input1_array.data_type) << ")."; in Run()
Dpropagate_fixed_sizes.cc485 const auto& input1_array = model->GetArray(op->inputs[1]); in ProcessSimpleBinaryOperator() local
487 if (!input0_array.has_shape() || !input1_array.has_shape()) { in ProcessSimpleBinaryOperator()
492 ComputeBinaryOperatorOutputSize(input0_array.shape(), input1_array.shape(), in ProcessSimpleBinaryOperator()
506 const auto& input1_array = model->GetArray(op->inputs[1]); in ProcessSelectOperator() local
508 output_array.copy_shape(input1_array.shape()); in ProcessSelectOperator()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dexport_test.cc61 Array& input1_array = input_model_.GetOrCreateArray(op->inputs[0]); in AddOperatorsByName() local
64 input1_array.data_type = ArrayDataType::kFloat; in AddOperatorsByName()
72 Array& input1_array = input_model_.GetOrCreateArray(op->inputs[0]); in AddOperatorsByName() local
75 input1_array.data_type = ArrayDataType::kFloat; in AddOperatorsByName()
78 input1_array.copy_shape({1, 2, 2, 2}); in AddOperatorsByName()
146 Array& input1_array = input_model_.GetOrCreateArray(op->inputs[0]); in BuildQuantizableTestModel() local
149 input1_array.data_type = ArrayDataType::kFloat; in BuildQuantizableTestModel()
Doperator.cc276 const Array& input1_array = op_signature.model->GetArray(input1_name); in GetVersion() local
280 if (input1_array.has_shape() && input2_array.has_shape()) { in GetVersion()
282 std::max(input1_array.shape().dimensions_count(), in GetVersion()
285 (input1_array.shape() != input2_array.shape()); in GetVersion()
313 const Array& input1_array = op_signature.model->GetArray(input1_name); in GetVersion() local
317 if (input1_array.has_shape() && input2_array.has_shape()) { in GetVersion()
319 std::max(input1_array.shape().dimensions_count(), in GetVersion()
322 (input1_array.shape() != input2_array.shape()); in GetVersion()
671 const Array& input1_array = op_signature.model->GetArray(input1_name); in GetVersion() local
674 const auto& input1_quant = input1_array.quantization_params; in GetVersion()