Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dbatch_to_space_nd.cc62 int spatial_dims_num = input_size->size - 2; in ResizeOutputTensor() local
H A Dspace_to_batch_nd.cc63 int spatial_dims_num = input_size->size - 2; in ResizeOutputTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A 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