Searched defs:block_shape_dims (Results 1 – 5 of 5) sorted by relevance
67 const std::vector<int>& block_shape_dims = block_shape_array.shape().dims(); in Run() local
71 const std::vector<int>& block_shape_dims = block_shape_array.shape().dims(); in Run() local
104 std::initializer_list<int> block_shape_dims = {2}, in SpaceToBatchNDOpDynamicModel()
2020 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()
4849 const Dims<4>& block_shape_dims, in BatchToSpaceND()