Home
last modified time | relevance | path

Searched refs:input2_tensor (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dxnnpack_delegate.cc1840 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 Darithmetic_builder.cc39 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 Dmin_max_builder_test.cc54 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 Dop_signature.cc166 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 Dlite_test.py1539 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 Dmodel_builder.cc1219 const TfLiteTensor* input2_tensor = reader->GetInputTensor(1); in Parse() local
1221 RETURN_IF_ERROR(ExtractTensorShape(*input2_tensor, &input2_shape)); in Parse()