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()
2099 std::vector<int64> block_shape_vec; in SpaceToBatchShapeHelper() local2101 block_shape_vec = GetFlatInt64(*block_shape_t); in SpaceToBatchShapeHelper()2103 const int64 block_shape_value = block_shape_vec[dim]; in SpaceToBatchShapeHelper()2134 TF_RETURN_IF_ERROR(c->Divide(padded_size, block_shape_vec[dim], in SpaceToBatchShapeHelper()2174 std::vector<int64> block_shape_vec; in BatchToSpaceShapeHelper() local2176 block_shape_vec = GetFlatInt64(*block_shape_t); in BatchToSpaceShapeHelper()2178 const int64 block_shape_value = block_shape_vec[dim]; in BatchToSpaceShapeHelper()2207 block_shape_vec[dim], &cropped_size)); in BatchToSpaceShapeHelper()2260 auto block_shape_vec = block_shape.vec<int64>(); in __anondb9326b23402() local2261 block_shape_vec(0) = block_size; in __anondb9326b23402()[all …]