Home
last modified time | relevance | path

Searched defs:block_shape_vec (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Darray_ops.cc2289 std::vector<int64_t> block_shape_vec; in SpaceToBatchShapeHelper() local
2364 std::vector<int64_t> block_shape_vec; in BatchToSpaceShapeHelper() local
2450 auto block_shape_vec = block_shape.vec<int64_t>(); in __anon38bbb0e83402() local
2490 auto block_shape_vec = block_shape.vec<int64_t>(); in __anon38bbb0e83602() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dspacetobatch_op.cc244 auto block_shape_vec = block_shape_.vec<int64_t>(); in SpaceToBatchOp() local
H A Dbatchtospace_op.cc227 auto block_shape_vec = block_shape_.vec<int64_t>(); in BatchToSpaceOp() local