Searched refs:block_shape_vec (Results 1 – 3 of 3) sorted by relevance
234 auto block_shape_vec = block_shape_.vec<int64>(); in SpaceToBatchOp() local235 block_shape_vec(0) = block_size_; in SpaceToBatchOp()236 block_shape_vec(1) = block_size_; in SpaceToBatchOp()
229 auto block_shape_vec = block_shape_.vec<int64>(); in BatchToSpaceOp() local230 block_shape_vec(0) = block_size_; in BatchToSpaceOp()231 block_shape_vec(1) = block_size_; in BatchToSpaceOp()
2256 std::vector<int64> block_shape_vec; in SpaceToBatchShapeHelper() local2258 block_shape_vec = GetFlatInt64(*block_shape_t); in SpaceToBatchShapeHelper()2260 const int64 block_shape_value = block_shape_vec[dim]; in SpaceToBatchShapeHelper()2291 TF_RETURN_IF_ERROR(c->Divide(padded_size, block_shape_vec[dim], in SpaceToBatchShapeHelper()2331 std::vector<int64> block_shape_vec; in BatchToSpaceShapeHelper() local2333 block_shape_vec = GetFlatInt64(*block_shape_t); in BatchToSpaceShapeHelper()2335 const int64 block_shape_value = block_shape_vec[dim]; in BatchToSpaceShapeHelper()2364 block_shape_vec[dim], &cropped_size)); in BatchToSpaceShapeHelper()2417 auto block_shape_vec = block_shape.vec<int64>(); in __anondb9326b23402() local2418 block_shape_vec(0) = block_size; in __anondb9326b23402()[all …]