HomeSort by relevance Sort by last modified time
    Searched defs:input_num_dims (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/lite/kernels/
squeeze.cc 44 int input_num_dims = NumDimensions(op_context.input); local
50 TF_LITE_ENSURE(context, input_num_dims <= 8);
54 for (int idx = 0; idx < input_num_dims; ++idx) {
62 int current = squeeze_dims[idx] < 0 ? squeeze_dims[idx] + input_num_dims
64 TF_LITE_ENSURE(context, current >= 0 && current < input_num_dims &&
72 TfLiteIntArrayCreate(input_num_dims - num_squeezed_dims);
73 for (int in_idx = 0, out_idx = 0; in_idx < input_num_dims; ++in_idx) {
mean.cc 72 int input_num_dims = NumDimensions(op_context->input); local
75 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(input_num_dims);
76 for (int idx = 0; idx < input_num_dims; ++idx) {
79 if (axis[axis_idx] == idx || axis[axis_idx] + input_num_dims == idx) {
97 current += input_num_dims;
99 TF_LITE_ENSURE(context, current >= 0 && current < input_num_dims);
103 previous += input_num_dims;
113 TfLiteIntArrayCreate(input_num_dims - num_reduce_axis);
115 for (int idx = 0; idx < input_num_dims; ++idx) {
118 if (axis[axis_idx] == idx || axis[axis_idx] + input_num_dims == idx)
    [all...]

Completed in 435 milliseconds