Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/delegates/hexagon/
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()
342 IsConstantTensor(GetInput(context, node, 1)); in IsNodeSupportedByHexagon()
364 IsConstantTensor(GetInput(context, node, 0)); in IsNodeSupportedByHexagon()
369 !IsConstantTensor(GetInput(context, node, 1))) { in IsNodeSupportedByHexagon()
408 if (!IsConstantTensor(&begins_tensor) || !IsConstantTensor(&sizes_tensor)) in IsNodeSupportedByHexagon()
434 if (!IsConstantTensor(&begins_tensor) || in IsNodeSupportedByHexagon()
435 !IsConstantTensor(&ends_tensor) || !IsConstantTensor(&step_tensor)) in IsNodeSupportedByHexagon()
/external/tensorflow/tensorflow/lite/kernels/
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()
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()
Drange.cc114 if (IsConstantTensor(start) && IsConstantTensor(limit) && in Prepare()
115 IsConstantTensor(delta)) { in Prepare()
Dsegment_sum.cc80 if (!IsConstantTensor(data) || !IsConstantTensor(segment_ids)) { in Prepare()
Dtranspose_conv.cc319 if (!IsConstantTensor(output_shape)) { in Prepare()
341 if (!IsConstantTensor(weights)) { in Prepare()
363 if (!IsConstantTensor(output_shape)) { in Prepare()
624 if (!IsConstantTensor(weights)) { in Eval()
642 if (!IsConstantTensor(weights)) { in Eval()
660 if (data->weights_are_transposed && !IsConstantTensor(weights)) { in Eval()
677 if (data->weights_are_transposed && !IsConstantTensor(weights)) { in Eval()
Dsplit.cc99 if (IsConstantTensor(op_context.axis)) { in Prepare()
112 if (!IsConstantTensor(op_context.axis)) { in Eval()
Dbatch_to_space_nd.cc106 if (!IsConstantTensor(op_context.block_shape) || in Prepare()
107 !IsConstantTensor(op_context.crops)) { in Prepare()
Dstrided_slice.cc257 if (!(IsConstantTensor(op_context.begin) && in Prepare()
258 IsConstantTensor(op_context.end) && in Prepare()
259 IsConstantTensor(op_context.strides))) { in Prepare()
Dspace_to_batch_nd.cc107 if (!IsConstantTensor(op_context.block_shape) || in Prepare()
108 !IsConstantTensor(op_context.paddings)) { in Prepare()
Dfully_connected.cc700 op_params.lhs_cacheable = IsConstantTensor(filter); in FullyConnectedInt8()
701 op_params.rhs_cacheable = IsConstantTensor(input); in FullyConnectedInt8()
776 op_params.lhs_cacheable = IsConstantTensor(filter); in EvalQuantized()
777 op_params.rhs_cacheable = IsConstantTensor(input); in EvalQuantized()
859 op_params.lhs_cacheable = IsConstantTensor(filter); in EvalShuffledQuantized()
860 op_params.rhs_cacheable = IsConstantTensor(input); in EvalShuffledQuantized()
948 op_params.lhs_cacheable = IsConstantTensor(filter); in EvalFloat()
949 op_params.rhs_cacheable = IsConstantTensor(input); in EvalFloat()
Dslice.cc140 if (!(IsConstantTensor(begin) && IsConstantTensor(size))) { in Prepare()
Dnumeric_verify.cc145 if (IsConstantTensor(op_context.input) && op_data->float_input_initialized) { in Eval()
160 if (IsConstantTensor(op_context.input)) { in Eval()
Dnon_max_suppression.cc103 const bool is_max_output_size_const = IsConstantTensor(input_max_output_size); in Prepare()
217 const bool is_max_output_size_const = IsConstantTensor(input_max_output_size); in Eval()
Dpad.cc58 if (IsConstantTensor(paddings) && paddings_total == 8 && in PadContext()
129 !IsConstantTensor(op_context.paddings)) { in Prepare()
Dwhere.cc79 if (!IsConstantTensor(cond_tensor)) { in Prepare()
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dutil.cc60 if (!((IsConstantTensor(input_0) && NumElements(input_0) == 1) || in IsBroadcastable()
61 (IsConstantTensor(input_1) && NumElements(input_1) == 1))) in IsBroadcastable()
Dadd_op_builder.cc73 if (IsConstantTensor(input_0) && NumElements(input_0) == 1) { in RegisterInputs()
76 } else if (IsConstantTensor(input_1) && NumElements(input_1) == 1) { in RegisterInputs()
Dmul_op_builder.cc79 if (IsConstantTensor(input_0) && NumElements(input_0) == 1) { in RegisterInputs()
82 } else if (IsConstantTensor(input_1) && NumElements(input_1) == 1) { in RegisterInputs()
Dconvolution_op_builder.cc333 if (!IsConstantTensor(weights)) { in IsConvolutionOpSupported()
341 if (!IsConstantTensor(GetInput(context, node, kOutputShapeTensor))) { in IsConvolutionOpSupported()
346 !IsConstantTensor(GetInput(context, node, kBiasTensor))) { in IsConvolutionOpSupported()
Dfully_connected_op_builder.cc163 if (!IsFloatType(weights->type) || !IsConstantTensor(weights)) { in IsFullyConnectedOpSupported()
175 if (!IsFloatType(bias->type) || !IsConstantTensor(bias)) { in IsFullyConnectedOpSupported()
Dresize_bilinear_op_builder.cc96 if (!IsConstantTensor(GetInput(context, node, kOutputSize))) { in IsResizeBilinearOpSupported()
/external/libtextclassifier/native/utils/tflite/
Dtoken_encoder.cc105 if (tflite::IsConstantTensor(&output_length)) { in Prepare()
129 if (!tflite::IsConstantTensor(&output_length)) { in Eval()
/external/armnn/delegate/src/
DElementwiseBinary.hpp276 if(tflite::IsConstantTensor(&tfLiteInputTensor0)) in VisitElementwiseBinaryOperator()
290 if(tflite::IsConstantTensor(&tfLiteInputTensor1)) in VisitElementwiseBinaryOperator()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dresize_bilinear_builder.cc36 if (!IsConstantTensor(&size_tensor)) { in PopulateSubGraph()

123