Home
last modified time | relevance | path

Searched refs:IsConstantTensor (Results 1 – 25 of 71) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/
H A Dutils.cc179 IsConstantTensor(GetInput(context, node, 1)); in IsNodeSupportedByHexagon()
184 !IsConstantTensor(GetInput(context, node, 1))) in IsNodeSupportedByHexagon()
199 IsConstantTensor(GetInput(context, node, 1))); in IsNodeSupportedByHexagon()
354 IsConstantTensor(GetInput(context, node, 1)); in IsNodeSupportedByHexagon()
376 IsConstantTensor(GetInput(context, node, 0)); in IsNodeSupportedByHexagon()
381 !IsConstantTensor(GetInput(context, node, 1))) { in IsNodeSupportedByHexagon()
417 if (!IsConstantTensor(&begins_tensor) || !IsConstantTensor(&sizes_tensor)) in IsNodeSupportedByHexagon()
443 if (!IsConstantTensor(&begins_tensor) || in IsNodeSupportedByHexagon()
444 !IsConstantTensor(&ends_tensor) || !IsConstantTensor(&step_tensor)) in IsNodeSupportedByHexagon()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dsplit_v.cc153 if (IsConstantTensor(op_context.size_splits) && in Prepare()
154 IsConstantTensor(op_context.axis)) { in Prepare()
167 if (!IsConstantTensor(op_context.axis) || in Eval()
168 !IsConstantTensor(op_context.size_splits)) { in Eval()
H A Ddequantize.cc70 if (IsConstantTensor(op_context.input)) { in Prepare()
81 if (IsConstantTensor(op_context.input) && in Eval()
92 if (IsConstantTensor(op_context.input)) { in Eval()
H A Drange.cc114 if (IsConstantTensor(start) && IsConstantTensor(limit) && in Prepare()
115 IsConstantTensor(delta)) { in Prepare()
H A Dtranspose_conv.cc320 if (!IsConstantTensor(output_shape)) { in Prepare()
342 if (!IsConstantTensor(weights)) { in Prepare()
364 if (!IsConstantTensor(output_shape)) { in Prepare()
654 if (!IsConstantTensor(weights)) { in Eval()
672 if (!IsConstantTensor(weights)) { in Eval()
690 if (data->weights_are_transposed && !IsConstantTensor(weights)) { in Eval()
707 if (data->weights_are_transposed && !IsConstantTensor(weights)) { in Eval()
H A Dfully_connected.cc804 op_params.lhs_cacheable = IsConstantTensor(filter); in FullyConnectedInt8()
805 op_params.rhs_cacheable = IsConstantTensor(input); in FullyConnectedInt8()
861 op_params.lhs_cacheable = IsConstantTensor(filter); in FullyConnectedPerChannelInt8()
862 op_params.rhs_cacheable = IsConstantTensor(input); in FullyConnectedPerChannelInt8()
984 op_params.lhs_cacheable = IsConstantTensor(filter); in EvalQuantized()
985 op_params.rhs_cacheable = IsConstantTensor(input); in EvalQuantized()
1134 op_params.lhs_cacheable = IsConstantTensor(filter); in EvalShuffledQuantized()
1135 op_params.rhs_cacheable = IsConstantTensor(input); in EvalShuffledQuantized()
1231 op_params.lhs_cacheable = IsConstantTensor(filter); in EvalFloat()
1232 op_params.rhs_cacheable = IsConstantTensor(input); in EvalFloat()
H A Dbatch_matmul.cc194 if (IsConstantTensor(op_context->rhs)) { in InitializeTemporaries()
533 op_params.lhs_cacheable = IsConstantTensor(lhs); in EvalInt8Int8()
534 op_params.rhs_cacheable = IsConstantTensor(rhs); in EvalInt8Int8()
571 op_params.lhs_cacheable = IsConstantTensor(lhs); in EvalInt8Int32()
572 op_params.rhs_cacheable = IsConstantTensor(rhs); in EvalInt8Int32()
729 if (!(IsConstantTensor(rhs) && op_data->rhs_transposed)) { in Eval()
H A Drandom_ops.cc188 if (!IsConstantTensor(shape)) { in Prepare()
217 if (!IsConstantTensor(logits) || !IsConstantTensor(num_samples)) { in PrepareMultinomial()
H A Dstrided_slice.cc257 if (!(IsConstantTensor(op_context.begin) && in Prepare()
258 IsConstantTensor(op_context.end) && in Prepare()
259 IsConstantTensor(op_context.strides))) { in Prepare()
H A Dsegment_sum.cc80 if (!IsConstantTensor(data) || !IsConstantTensor(segment_ids)) { in Prepare()
H A Dsplit.cc99 if (IsConstantTensor(op_context.axis)) { in Prepare()
112 if (!IsConstantTensor(op_context.axis)) { in Eval()
H A Dbatch_to_space_nd.cc106 if (!IsConstantTensor(op_context.block_shape) || in Prepare()
107 !IsConstantTensor(op_context.crops)) { in Prepare()
H A Dspace_to_batch_nd.cc107 if (!IsConstantTensor(op_context.block_shape) || in Prepare()
108 !IsConstantTensor(op_context.paddings)) { in Prepare()
H A Dkernel_util.h193 inline bool IsConstantTensor(const TfLiteTensor* tensor) { in IsConstantTensor() function
198 return IsConstantTensor(tensor) || in IsConstantOrPersistentTensor()
H A Dnumeric_verify.cc145 if (IsConstantTensor(op_context.input) && op_data->float_input_initialized) { in Eval()
160 if (IsConstantTensor(op_context.input)) { in Eval()
H A Dslice.cc153 if (!(IsConstantTensor(begin) && IsConstantTensor(size)) || in Prepare()
H A Dunsorted_segment.cc109 if (IsDynamicTensor(data) || !IsConstantTensor(segment_ids) || in Prepare()
110 !IsConstantTensor(num_segments)) { in Prepare()
/aosp_15_r20/external/armnn/delegate/opaque/src/
H A DOpaqueDelegateUtils.hpp108 bool IsConstantTensor(const TfLiteOpaqueTensor* tfLiteTensor) in IsConstantTensor() function
454 if(IsConstantTensor(tfLiteTensor)) in GetTensorInfoForTfLiteOpaqueTensor()
485 if(IsConstantTensor(tfLiteTensor)) in GetTensorInfoForTfLiteOpaqueTensor()
596 if (IsConstantTensor(tfLiteInputTensor)) in ProcessInputs()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
H A Dutil.cc60 if (!((IsConstantTensor(input_0) && NumElements(input_0) == 1) || in IsBroadcastable()
61 (IsConstantTensor(input_1) && NumElements(input_1) == 1))) in IsBroadcastable()
H A Dconvolution_op_builder.cc337 if (!IsConstantTensor(weights)) { in IsConvolutionOpSupported()
345 if (!IsConstantTensor(GetInput(context, node, kOutputShapeTensor))) { in IsConvolutionOpSupported()
350 !IsConstantTensor(GetInput(context, node, kBiasTensor))) { in IsConvolutionOpSupported()
H A Dmul_op_builder.cc82 if (IsConstantTensor(input_0) && NumElements(input_0) == 1) { in RegisterInputs()
85 } else if (IsConstantTensor(input_1) && NumElements(input_1) == 1) { in RegisterInputs()
H A Dadd_op_builder.cc76 if (IsConstantTensor(input_0) && NumElements(input_0) == 1) { in RegisterInputs()
79 } else if (IsConstantTensor(input_1) && NumElements(input_1) == 1) { in RegisterInputs()
H A Dfully_connected_op_builder.cc167 if (!IsFloatType(weights->type) || !IsConstantTensor(weights)) { in IsFullyConnectedOpSupported()
179 if (!IsFloatType(bias->type) || !IsConstantTensor(bias)) { in IsFullyConnectedOpSupported()
/aosp_15_r20/external/armnn/delegate/classic/src/
H A DClassicDelegateUtils.hpp378 if(tflite::IsConstantTensor(&tfLiteTensor)) in GetTensorInfoForTfLiteTensor()
405 if(tflite::IsConstantTensor(&tfLiteTensor)) in GetTensorInfoForTfLiteTensor()
489 if (tflite::IsConstantTensor(&tfLiteInputTensor)) in ProcessInputs()
/aosp_15_r20/external/libtextclassifier/native/utils/tflite/
H A Dtoken_encoder.cc105 if (tflite::IsConstantTensor(&output_length)) { in Prepare()
129 if (!tflite::IsConstantTensor(&output_length)) { in Eval()

123