| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | pad_test.cc | 90 PaddingConfig padding_config; in XLA_TEST_P() local
105 PaddingConfig padding_config; in XLA_TEST_P() local
120 PaddingConfig padding_config; in XLA_TEST_P() local
188 PaddingConfig padding_config; in TEST_P() local
230 PaddingConfig padding_config; in XLA_TEST_P() local
327 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
349 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
373 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
400 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
428 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling.cc | 54 PaddingConfig padding_config; in AvgPoolDivideByCountWithGeneralPadding() local
93 PaddingConfig padding_config; in MakeSpatialPaddingConfig() local
162 auto padding_config = MakeSpatialPaddingConfig(padding, num_spatial_dims, in AvgPool() local
242 PaddingConfig padding_config = MakeNoPaddingConfig(num_dims); in AvgPoolGrad() local
|
| D | matrix.cc | 192 PaddingConfig padding_config; in SetMatrixDiagonal() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_pad_op.cc | 72 xla::PaddingConfig padding_config; in Compile() local
|
| D | slice_op.cc | 162 xla::PaddingConfig padding_config; in Compile() local
|
| D | matrix_diag_ops.cc | 128 xla::PaddingConfig padding_config; in SetMatrixDiag() local
402 xla::PaddingConfig padding_config = in Compile() local
|
| D | spacetobatch_op.cc | 53 xla::PaddingConfig padding_config; in SpaceToBatch() local
|
| D | dynamic_partition_op.cc | 90 xla::PaddingConfig padding_config; in DynamicPartition1D() local
|
| D | strided_slice_op.cc | 112 xla::PaddingConfig padding_config; in EmitDynamicSlice() local
570 xla::PaddingConfig padding_config; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_padding_legalization.cc | 62 PaddingConfig padding_config = in MaybePaddedAndSlicedInput() local
131 PaddingConfig padding_config; in MaybePaddedKernel() local
|
| D | tree_reduction_rewriter.cc | 121 PaddingConfig padding_config = MakeNoPaddingConfig(input_shape.rank()); in RewriteReduction() local
|
| D | cublas_pad_for_gemms.cc | 69 PaddingConfig padding_config; in PadForGemm() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | util.cc | 156 PaddingConfig padding_config; in MakeNoPaddingConfig() local
168 PaddingConfig padding_config; in MakeEdgePaddingConfig() local
|
| D | window_util.cc | 162 bool HasSymmetricPadding(const PaddingConfig& padding_config) { in HasSymmetricPadding()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | mlir_hlo_builder_test.cc | 161 PaddingConfig padding_config; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 70 const PaddingConfig& padding_config) { in MakePadHlo()
541 PaddingConfig padding_config; in PadVectorWithZeros() local
|
| D | space_to_batch_converter.cc | 464 PaddingConfig padding_config = in HaloDuplicateWithSlice() local
659 PaddingConfig padding_config = in IncreaseSpatialSizeOnSpaceToBatchedShape() local
1612 PaddingConfig padding_config = in Propagate() local
1913 PaddingConfig padding_config = in Propagate() local
2432 PaddingConfig padding_config = in SplitSpaceHelper() local
|
| D | triangular_solve_expander.cc | 70 PaddingConfig padding_config = in DiagonalBlocks() local
|
| D | hlo.proto | 147 xla.PaddingConfig padding_config = 21; field
|
| D | hlo_verifier_test.cc | 358 PaddingConfig padding_config; in TEST_F() local
382 PaddingConfig padding_config; in TEST_F() local
|
| D | hlo_instructions.h | 1652 const PaddingConfig& padding_config() const { return padding_config_; } in padding_config() function
|
| D | algebraic_simplifier.cc | 1378 PaddingConfig padding_config; in HandleConcatenate() local
3422 PaddingConfig padding_config = pad->padding_config(); in HandlePad() local
4673 PaddingConfig padding_config; in HandleDynamicUpdateSlice() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 421 const auto& padding_config = instruction->padding_config(); in ImportInstructionImpl() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 558 const PaddingConfig& padding_config) { in PadInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 256 PaddingConfig padding_config; in PadToShape() local
1012 PaddingConfig padding_config; in ExchangeHaloAndGetValidData() local
|