Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatchtospace_op.cc52 int64 block_num_elems = 1; in BatchToSpace() local
Dspacetobatch_op.cc55 int64 block_num_elems = 1LL; in SpaceToBatch() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc830 int block_num_elems = 1; in convertSpaceToBatchNDOp() local
1100 int block_num_elems = 1; in convertBatchToSpaceNDOp() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc1061 int64_t block_num_elems = 1; in matchAndRewrite() local