| /aosp_15_r20/external/ComputeLibrary/src/graph/mutators/ |
| H A D | InPlaceOperationMutator.cpp | 163 auto input1_tensor = input1_edge->tensor(); in try_in_place_elementwise() local
164 ARM_COMPUTE_ERROR_ON(input0_tensor == nullptr || input1_tensor == nullptr); in try_in_place_elementwise()
167 const auto shape1 = input1_tensor->desc().shape; in try_in_place_elementwise()
169 const auto qinfo1 = input1_tensor->desc().quant_info; in try_in_place_elementwise()
187 …&& (input1_tensor->desc().data_type == current_output_tensor->desc().data_type) && (input1_tensor-… in try_in_place_elementwise()
195 set_new_output_and_inherit_accessor(node, current_output_tensor, input1_tensor); in try_in_place_elementwise()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/testing/op_tests/ |
| H A D | einsum.py | 69 input1_tensor = tf.compat.v1.placeholder(
71 out = tf.einsum(equation, input0_tensor, input1_tensor)
72 return [input0_tensor, input1_tensor], [out]
|
| H A D | batchmatmul.py | 72 input1_tensor = tf.compat.v1.placeholder(
75 input0_tensor, input1_tensor, adjoint_a=adj_a, adjoint_b=adj_b)
76 return [input0_tensor, input1_tensor], [out]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/ |
| H A D | einsum.py | 65 input1_tensor = tf.compat.v1.placeholder(
67 out = tf.einsum(equation, input0_tensor, input1_tensor)
68 return [input0_tensor, input1_tensor], [out]
|
| H A D | batchmatmul.py | 74 input1_tensor = tf.compat.v1.placeholder(
77 input0_tensor, input1_tensor, adjoint_a=adj_a, adjoint_b=adj_b)
78 return [input0_tensor, input1_tensor], [out]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| H A D | xnnpack_delegate.cc | 1839 TfLiteContext* context, const TfLiteTensor& input1_tensor, in CheckTensorsInputProductOutputScale() argument
1842 if (input1_tensor.type != input2_tensor.type || in CheckTensorsInputProductOutputScale()
1843 input1_tensor.type != output_tensor.type) { in CheckTensorsInputProductOutputScale()
1848 if (input1_tensor.type == kTfLiteInt8 || in CheckTensorsInputProductOutputScale()
1849 input1_tensor.type == kTfLiteUInt8) { in CheckTensorsInputProductOutputScale()
1851 input1_tensor.quantization.params) in CheckTensorsInputProductOutputScale()
2187 const TfLiteTensor& input1_tensor = tensors[node->inputs->data[0]]; in VisitAddNode() local
2189 CheckTensorFloat32OrQUInt8Type(delegate, logging_context, input1_tensor, in VisitAddNode()
2192 logging_context, input1_tensor, node->inputs->data[0], node_index)); in VisitAddNode()
2211 logging_context, input1_tensor, output_tensor, scale_min, scale_max, in VisitAddNode()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| H A D | hardswish_builder.cc | 32 const auto& input1_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local
34 TF_LITE_ENSURE_STATUS(ComputeAndAddMinAndMax(context, input1_tensor)); in PopulateSubGraph()
|
| H A D | arithmetic_builder.cc | 34 const auto& input1_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local
43 TF_LITE_ENSURE_STATUS(ComputeAndAddMinAndMax(context, input1_tensor)); in PopulateSubGraph()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| H A D | min_max_builder_test.cc | 51 auto* input1_tensor = interpreter_->tensor(input1_); in MinMaxOpModel() local
52 input1_tensor->allocation_type = kTfLiteMmapRo; in MinMaxOpModel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/versioning/ |
| H A D | op_signature.cc | 164 const Tensor* input1_tensor = in GetOpSignature() local
171 input1_tensor->quantization(); in GetOpSignature()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/ |
| H A D | interpreter_test.cc | 1817 const TfLiteTensor* input1_tensor = in TEST_F() local
1820 NewCustomAlloc(2 * input1_tensor->bytes, kDefaultTensorAlignment); in TEST_F()
|