Home
last modified time | relevance | path

Searched refs:remainder_shape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dspacetobatch_op.cc37 absl::Span<const int64> remainder_shape(input_shape); in SpaceToBatch() local
38 remainder_shape.remove_prefix(1 + block_rank); in SpaceToBatch()
69 for (int i = 0; i < remainder_shape.size(); ++i) { in SpaceToBatch()
101 std::copy(remainder_shape.begin(), remainder_shape.end(), in SpaceToBatch()
141 std::copy(remainder_shape.begin(), remainder_shape.end(), in SpaceToBatch()
Dbatchtospace_op.cc37 absl::Span<const int64> remainder_shape(input_shape); in BatchToSpace() local
38 remainder_shape.remove_prefix(1 + block_rank); in BatchToSpace()
110 std::copy(remainder_shape.begin(), remainder_shape.end(), in BatchToSpace()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc1056 auto remainder_shape = input_shape.drop_front(1 + block_rank); in matchAndRewrite() local
1137 std::copy(remainder_shape.begin(), remainder_shape.end(), in matchAndRewrite()