Searched refs:TF_LITE_ENSURE_MSG (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/contrib/lite/kernels/ |
D | strided_slice.cc | 117 TF_LITE_ENSURE_MSG(context, stride != 0, "stride value has to be non-zero"); in ResizeOutputTensor() 158 TF_LITE_ENSURE_MSG(context, op_context.dims <= 4, in Prepare() 162 TF_LITE_ENSURE_MSG(context, op_context.params->ellipsis_mask == 0, in Prepare() 164 TF_LITE_ENSURE_MSG(context, op_context.params->new_axis_mask == 0, in Prepare()
|
D | transpose.cc | 54 TF_LITE_ENSURE_MSG(context, (perm_data[idx] >= 0 && perm_data[idx] < dims), in ResizeOutputTensor() 75 TF_LITE_ENSURE_MSG(context, NumDimensions(op_context.input) <= 4, in Prepare()
|
D | topk_v2.cc | 44 TF_LITE_ENSURE_MSG(context, input->dims->size >= 1, in ResizeOutput() 47 TF_LITE_ENSURE_MSG(context, k <= input->dims->data[num_dimensions - 1], in ResizeOutput()
|
D | split.cc | 57 TF_LITE_ENSURE_MSG(context, input_size % num_splits == 0, in ResizeOutputTensors()
|
D | pad.cc | 68 TF_LITE_ENSURE_MSG(context, (before_padding >= 0 && after_padding >= 0), in ResizeOutputTensor()
|
/external/tensorflow/tensorflow/contrib/lite/ |
D | context.h | 80 #define TF_LITE_ENSURE_MSG(context, value, msg) \ macro
|