Searched refs:low_padding (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | pad_test.cc | 367 constexpr int64 low_padding = 0; in XLA_TEST_P() local
375 padding_config.mutable_dimensions(dim)->set_edge_padding_low(low_padding); in XLA_TEST_P()
394 int64 low_padding[2] = {-1, -2}; in XLA_TEST_P() local
403 low_padding[dim]); in XLA_TEST_P()
422 int64 low_padding[2] = {4, -1}; in XLA_TEST_P() local
431 low_padding[dim]); in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 119 int64 high_padding, int64 low_padding, int64 spatial_split_size,
126 int64 activations_batch_dim, int64 high_padding, int64 low_padding,
178 int64 activations_batch_dim, int64 old_batch_size, int64 low_padding,
345 int64 activations_batch_dim, int64 old_batch_size, int64 low_padding, in HaloDuplicateWithSlice() argument
372 CHECK_LE(std::abs(halo_size - low_padding), spatial_split_size); in HaloDuplicateWithSlice()
376 << low_padding << " halo size " << halo_size; in HaloDuplicateWithSlice()
387 if (low_padding > 0) { in HaloDuplicateWithSlice()
392 spatial_split_size - low_padding; in HaloDuplicateWithSlice()
409 if (halo_size - low_padding > 0) { in HaloDuplicateWithSlice()
415 end_indices_halo[spatial_dimension_to_split] = halo_size - low_padding; in HaloDuplicateWithSlice()
[all …]
|
| D | algebraic_simplifier_test.cc | 3282 int64 low_padding[2] = {-1, -2}; in TEST_F() local
3286 dimension->set_edge_padding_low(low_padding[i]); in TEST_F()
3330 int64 low_padding[2] = {-1, -2}; in TEST_F() local
3334 dimension->set_edge_padding_low(low_padding[i]); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 1085 int64 low_padding = 0; in HaloExchangeToPadOnLeft() local
1087 low_padding = in HaloExchangeToPadOnLeft()
1092 dim->set_padding_low(low_padding); in HaloExchangeToPadOnLeft()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2411 DimVector low_padding(num_dims, 0); in matchAndRewrite() local
2444 low_padding[dim] = in matchAndRewrite()
2450 out_grad_shape[dim] = low_padding[dim] + high_padding[dim] + in matchAndRewrite()
2457 /*edge_padding_low=*/GetI64ElementsAttr(low_padding, &rewriter), in matchAndRewrite()
5434 llvm::SmallVector<int64_t, 4> low_padding(rank, 0); in matchAndRewrite() local
5437 low_padding[axis] = 1; in matchAndRewrite()
5441 GetI64ElementsAttr(low_padding, &rewriter), in matchAndRewrite()
|