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()
|