Home
last modified time | relevance | path

Searched refs:TF_LITE_ENSURE_MSG (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/
Dbroadcast_to.cc55 TF_LITE_ENSURE_MSG(context, input_num_dims <= output_num_dims, in ResizeOutputTensor()
57 TF_LITE_ENSURE_MSG(context, output_num_dims <= kMaxDims, in ResizeOutputTensor()
71 TF_LITE_ENSURE_MSG(context, in ResizeOutputTensor()
92 TF_LITE_ENSURE_MSG(context, in Prepare()
Dstrided_slice.cc110 TF_LITE_ENSURE_MSG(context, stride != 0, "stride value has to be non-zero"); in ResizeOutputTensor()
162 TF_LITE_ENSURE_MSG(context, op_context.dims <= 5, in Prepare()
169 TF_LITE_ENSURE_MSG(context, op_context.params->ellipsis_mask == 0, in Prepare()
171 TF_LITE_ENSURE_MSG(context, op_context.params->new_axis_mask == 0, in Prepare()
Dtranspose.cc57 TF_LITE_ENSURE_MSG(context, (perm_data[idx] >= 0 && perm_data[idx] < dims), in ResizeOutputTensor()
78 TF_LITE_ENSURE_MSG(context, NumDimensions(op_context.input) <= 5, in Prepare()
Dtopk_v2.cc50 TF_LITE_ENSURE_MSG(context, input->dims->size >= 1, in ResizeOutput()
53 TF_LITE_ENSURE_MSG(context, k <= input->dims->data[num_dimensions - 1], in ResizeOutput()
Dsplit.cc63 TF_LITE_ENSURE_MSG(context, input_size % num_splits == 0, in ResizeOutputTensors()
Dslice.cc135 TF_LITE_ENSURE_MSG(context, NumDimensions(input) <= kMaxDim, in Prepare()
Dpad.cc89 TF_LITE_ENSURE_MSG(context, (before_padding >= 0 && after_padding >= 0), in ResizeOutputTensor()
Delementwise.cc244 TF_LITE_ENSURE_MSG(context, i >= op_data->input_offset, in RsqrtEvalQuantized()
/external/tensorflow/tensorflow/lite/kernels/perception/
Ddense_image_warp.cc113 TF_LITE_ENSURE_MSG(context, input_shape.Dims(1) >= 2, in Prepare()
115 TF_LITE_ENSURE_MSG(context, input_shape.Dims(2) >= 2, in Prepare()
117 TF_LITE_ENSURE_MSG(context, flow_shape.Dims(3) == 2, in Prepare()
Dmax_unpooling_2d.cc77 TF_LITE_ENSURE_MSG( in Prepare()
81 TF_LITE_ENSURE_MSG(context, input_shape.Dims(i) == indices_shape.Dims(i), in Prepare()
Dmax_pool_with_argmax.cc172 TF_LITE_ENSURE_MSG( in Prepare()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dtranspose.cc57 TF_LITE_ENSURE_MSG(context, (perm_data[idx] >= 0 && perm_data[idx] < dims), in ResizeOutputTensor()
78 TF_LITE_ENSURE_MSG(context, NumDimensions(op_context.input) <= 5, in Prepare()
Dstrided_slice.cc95 TF_LITE_ENSURE_MSG(context, stride != 0, "stride value has to be non-zero"); in CheckOutputSize()
132 TF_LITE_ENSURE_MSG(context, op_context.dims <= kMaxDim, in Prepare()
Dsplit.cc77 TF_LITE_ENSURE_MSG(context, IsConstantTensor(axis), in Prepare()
Dsplit_v.cc81 TF_LITE_ENSURE_MSG(context, IsConstantTensor(axis), in Prepare()
Dconv.cc56 TF_LITE_ENSURE_MSG(context, input->type == filter->type, in Eval()
Dfully_connected.cc57 TF_LITE_ENSURE_MSG(context, input->type == filter->type, in Prepare()
Dtranspose_conv.cc215 TF_LITE_ENSURE_MSG(context, input->type == filter->type, in Eval()
Dreduce.cc254 TF_LITE_ENSURE_MSG(context, false, in EvalMean()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/
Dsentencepiece_detokenizer_tflite.cc78 TF_LITE_ENSURE_MSG(context, res.type == DecoderResultType::SUCCESS, in Eval()
Dsentencepiece_tokenizer_tflite.cc92 TF_LITE_ENSURE_MSG(context, res.type == EncoderResultType::SUCCESS, in Eval()
/external/tensorflow/tensorflow/lite/c/
Dcommon.h173 #define TF_LITE_ENSURE_MSG(context, value, msg) \ macro
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dfully_connected.cc65 TF_LITE_ENSURE_MSG(context, input->type == filter->type, in Prepare()
Dconv.cc261 TF_LITE_ENSURE_MSG(context, input->type == filter->type, in Eval()
/external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/
Dlayer_norm.cc321 TF_LITE_ENSURE_MSG(context, false, in Eval()

12