Searched refs:input2_tensor (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
H A D | xnnpack_delegate.cc | 1840 const TfLiteTensor& input2_tensor, const TfLiteTensor& output_tensor, in CheckTensorsInputProductOutputScale() argument 1842 if (input1_tensor.type != input2_tensor.type || in CheckTensorsInputProductOutputScale() 1854 input2_tensor.quantization.params) in CheckTensorsInputProductOutputScale() 2194 const TfLiteTensor& input2_tensor = tensors[node->inputs->data[1]]; in VisitAddNode() local 2196 CheckTensorFloat32OrQUInt8Type(delegate, logging_context, input2_tensor, in VisitAddNode() 2199 logging_context, input2_tensor, node->inputs->data[1], node_index)); in VisitAddNode() 2214 logging_context, input2_tensor, output_tensor, scale_min, scale_max, in VisitAddNode() 2871 const TfLiteTensor& input2_tensor = tensors[node->inputs->data[1]]; in VisitDivNode() local 2873 logging_context, input2_tensor, node->inputs->data[1], node_index)); in VisitDivNode() 2875 logging_context, input2_tensor, node->inputs->data[1], node_index)); in VisitDivNode() [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
H A D | arithmetic_builder.cc | 39 const auto& input2_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local 44 TF_LITE_ENSURE_STATUS(ComputeAndAddMinAndMax(context, input2_tensor)); in PopulateSubGraph()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
H A D | min_max_builder_test.cc | 54 auto* input2_tensor = interpreter_->tensor(input2_); in MinMaxOpModel() local 55 input2_tensor->allocation_type = kTfLiteMmapRo; in MinMaxOpModel()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/versioning/ |
H A D | op_signature.cc | 166 const Tensor* input2_tensor = in GetOpSignature() local 172 const QuantizationParameters* input2_qunt = input2_tensor->quantization(); in GetOpSignature()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/ |
H A D | lite_test.py | 1539 input2_tensor = array_ops.placeholder(shape=[1], dtype=dtypes.float32) 1543 neg = math_ops.negative(input2_tensor) 1550 sess, [input_tensor, padding, input2_tensor], [output_tensor])
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
H A D | model_builder.cc | 1219 const TfLiteTensor* input2_tensor = reader->GetInputTensor(1); in Parse() local 1221 RETURN_IF_ERROR(ExtractTensorShape(*input2_tensor, &input2_shape)); in Parse()
|