Searched defs:input_num_dims (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | squeeze.cc | 43 int input_num_dims = NumDimensions(op_context.input); in Prepare() local
|
| D | reduce.cc | 95 int input_num_dims = NumDimensions(op_context->input); in ResizeOutputTensor() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 3434 const int* output_dims, const int input_num_dims, in Reduce()
3486 const int* output_dims, const int input_num_dims, in ReduceSumImpl()
3521 const int input_num_dims, T* output_data, in ReduceGeneric()
3550 const int input_num_dims, T* output_data, in Mean()
3712 const int input_num_dims, T* output_data, in QuantizedMeanOrSum()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 1721 const int input_num_dims = input_dims.size(); in ProcessSqueezeOperator() local
|