| /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 …]
|
| D | dynamism_inference_test.cc | 343 PaddingConfig padding_config; in TEST_F() local
|
| /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 | 191 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 | 153 xla::PaddingConfig padding_config; in Compile() local
|
| D | spacetobatch_op.cc | 53 xla::PaddingConfig padding_config; in SpaceToBatch() local
|
| D | matrix_diag_ops.cc | 126 xla::PaddingConfig padding_config; in SetMatrixDiag() local
400 xla::PaddingConfig padding_config = in Compile() local
|
| D | strided_slice_op.cc | 110 xla::PaddingConfig padding_config; in EmitDynamicSlice() local
532 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 | cublas_gemm_pad_for_tensor_cores.cc | 68 PaddingConfig padding_config; in PadForTensorCores() local
|
| D | tree_reduction_rewriter.cc | 125 PaddingConfig padding_config = MakeNoPaddingConfig(input_shape.rank()); in RewriteReduction() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | util.cc | 134 PaddingConfig padding_config; in MakeNoPaddingConfig() local
146 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 | space_to_batch_converter.cc | 399 PaddingConfig padding_config = in HaloDuplicateWithSlice() local
591 PaddingConfig padding_config = in IncreaseSpatialSizeOnSpaceToBatchedShape() local
1293 PaddingConfig padding_config = in Propagate() local
1560 PaddingConfig padding_config = in Propagate() local
2047 PaddingConfig padding_config = in SplitSpaceHelper() local
|
| D | hlo_creation_utils.cc | 70 const PaddingConfig& padding_config) { in MakePadHlo()
537 PaddingConfig padding_config; in PadVectorWithZeros() local
|
| D | hlo_verifier_test.cc | 358 PaddingConfig padding_config; in TEST_F() local
382 PaddingConfig padding_config; in TEST_F() local
|
| D | triangular_solve_expander.cc | 70 PaddingConfig padding_config = in DiagonalBlocks() local
|
| D | hlo.proto | 112 xla.PaddingConfig padding_config = 21; field
|
| D | hlo_instructions.h | 1551 const PaddingConfig& padding_config() const { return padding_config_; } in padding_config() function
|
| D | algebraic_simplifier.cc | 1327 PaddingConfig padding_config; in HandleConcatenate() local
3332 PaddingConfig padding_config = pad->padding_config(); in HandlePad() local
4453 PaddingConfig padding_config; in HandleDynamicUpdateSlice() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 409 const auto& padding_config = instruction->padding_config(); in ImportInstructionImpl() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 541 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
966 PaddingConfig padding_config; in ExchangeHaloAndGetValidData() local
|