Searched refs:remainder_shape (Results 1 – 3 of 3) sorted by relevance
37 absl::Span<const int64> remainder_shape(input_shape); in SpaceToBatch() local38 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()
37 absl::Span<const int64> remainder_shape(input_shape); in BatchToSpace() local38 remainder_shape.remove_prefix(1 + block_rank); in BatchToSpace()110 std::copy(remainder_shape.begin(), remainder_shape.end(), in BatchToSpace()
1056 auto remainder_shape = input_shape.drop_front(1 + block_rank); in matchAndRewrite() local1137 std::copy(remainder_shape.begin(), remainder_shape.end(), in matchAndRewrite()