Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A 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()
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tflite/
H A Dexport_test.cc66 Array& input1_array = input_model_.GetOrCreateArray(op->inputs[0]); in AddOperatorsByName() local
69 input1_array.data_type = ArrayDataType::kFloat; in AddOperatorsByName()
77 Array& input1_array = input_model_.GetOrCreateArray(op->inputs[0]); in AddOperatorsByName() local
80 input1_array.data_type = ArrayDataType::kFloat; in AddOperatorsByName()
83 input1_array.copy_shape({1, 2, 2, 2}); in AddOperatorsByName()
151 Array& input1_array = input_model_.GetOrCreateArray(op->inputs[0]); in BuildQuantizableTestModel() local
154 input1_array.data_type = ArrayDataType::kFloat; in BuildQuantizableTestModel()
H A Doperator.cc657 const Array& input1_array = op_signature.model->GetArray(input1_name); in GetVersion() local
660 const auto& input1_quant = input1_array.quantization_params; in GetVersion()