Home
last modified time | relevance | path

Searched defs:block_shape_dims (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_batch_to_space_nd_attributes.cc67 const std::vector<int>& block_shape_dims = block_shape_array.shape().dims(); in Run() local
Dresolve_space_to_batch_nd_attributes.cc71 const std::vector<int>& block_shape_dims = block_shape_array.shape().dims(); in Run() local
/external/tensorflow/tensorflow/lite/kernels/
Dspace_to_batch_nd_test.cc104 std::initializer_list<int> block_shape_dims = {2}, in SpaceToBatchNDOpDynamicModel()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h2020 const Dims<4>& block_shape_dims, in SpaceToBatchND()
2037 const Dims<4>& block_shape_dims, in SpaceToBatchND()
2053 const Dims<4>& block_shape_dims, in BatchToSpaceND()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h4849 const Dims<4>& block_shape_dims, in BatchToSpaceND()