Searched refs:block_shape_t (Results 1 – 1 of 1) sorted by relevance
2235 const Tensor* block_shape_t, in SpaceToBatchShapeHelper() argument2257 if (block_shape_t && (block_shape_t->NumElements() > 0)) { in SpaceToBatchShapeHelper()2258 block_shape_vec = GetFlatInt64(*block_shape_t); in SpaceToBatchShapeHelper()2286 if (block_shape_t) { in SpaceToBatchShapeHelper()2311 const Tensor* block_shape_t, in BatchToSpaceShapeHelper() argument2332 if (block_shape_t) { in BatchToSpaceShapeHelper()2333 block_shape_vec = GetFlatInt64(*block_shape_t); in BatchToSpaceShapeHelper()2361 if (block_shape_t) { in BatchToSpaceShapeHelper()