Home
last modified time | relevance | path

Searched defs:block_shape_value (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dspacetobatch_op.cc166 const int64_t block_shape_value = block_shape[block_dim]; in SpaceToBatchOpCompute() local
H A Dbatchtospace_op.cc155 const int64_t block_shape_value = block_shape[block_dim]; in BatchToSpaceOpCompute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Darray_ops.cc2293 const int64_t block_shape_value = block_shape_vec[dim]; in SpaceToBatchShapeHelper() local
2368 const int64_t block_shape_value = block_shape_vec[dim]; in BatchToSpaceShapeHelper() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_common.cc672 Value block_shape_value, in convertSpaceToBatchNDOp()
942 Value block_shape_value, in convertBatchToSpaceNDOp()