Home
last modified time | relevance | path

Searched defs:spatial_dims_num (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dbatch_to_space_nd.cc62 int spatial_dims_num = input_size->size - 2; in ResizeOutputTensor() local
Dspace_to_batch_nd.cc63 int spatial_dims_num = input_size->size - 2; in ResizeOutputTensor() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc1303 int spatial_dims_num = input_shape.dimensions_count() - 2; in ProcessSpaceToBatchNDOperator() local
1348 int spatial_dims_num = input_shape.dimensions_count() - 2; in ProcessBatchToSpaceNDOperator() local