Home
last modified time | relevance | path

Searched defs:padding_config (Results 1 – 25 of 43) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dpad_test.cc88 PaddingConfig padding_config; in XLA_TEST_P() local
103 PaddingConfig padding_config; in XLA_TEST_P() local
118 PaddingConfig padding_config; in XLA_TEST_P() local
186 PaddingConfig padding_config; in TEST_P() local
228 PaddingConfig padding_config; in XLA_TEST_P() local
325 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
347 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
371 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
398 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
426 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc54 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
Dmatrix.cc192 PaddingConfig padding_config; in SetMatrixDiagonal() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_simplify_padding.cc97 const PaddingConfig::PaddingConfigDimension& padding_config = in NumTrailingZeroOutputFeatures() local
144 const PaddingConfig::PaddingConfigDimension& padding_config = in NumTrailingZeroOutputFeatures() local
Dgpu_conv_padding_legalization.cc63 PaddingConfig padding_config = in MaybePaddedAndSlicedInput() local
133 PaddingConfig padding_config; in MaybePaddedKernel() local
Dcublas_pad_for_gemms.cc69 PaddingConfig padding_config; in PadForGemm() local
Dtree_reduction_rewriter.cc136 PaddingConfig padding_config = in RewriteReduction() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_pad_op.cc72 xla::PaddingConfig padding_config; in Compile() local
Dslice_op.cc162 xla::PaddingConfig padding_config; in Compile() local
Dmatrix_diag_ops.cc129 xla::PaddingConfig padding_config; in SetMatrixDiag() local
403 xla::PaddingConfig padding_config = in Compile() local
Dspacetobatch_op.cc54 xla::PaddingConfig padding_config; in SpaceToBatch() local
Ddynamic_partition_op.cc90 xla::PaddingConfig padding_config; in DynamicPartition1D() local
Dwhere_op.cc128 xla::PaddingConfig padding_config; in PrefixSum() local
Dstrided_slice_op.cc110 xla::PaddingConfig padding_config; in EmitDynamicSlice() local
571 xla::PaddingConfig padding_config; in Compile() local
Dsharding_util_ops.cc165 xla::PaddingConfig padding_config; in CompileInternal() local
/external/tensorflow/tensorflow/compiler/xla/
Dutil.cc194 PaddingConfig padding_config; in MakeNoPaddingConfig() local
206 PaddingConfig padding_config; in MakeEdgePaddingConfig() local
Dwindow_util.cc162 bool HasSymmetricPadding(const PaddingConfig& padding_config) { in HasSymmetricPadding()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h179 PaddingConfig padding_config; variable
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dmlir_hlo_builder_test.cc171 PaddingConfig padding_config; in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc564 PaddingConfig padding_config = in HaloDuplicateWithSlice() local
839 PaddingConfig padding_config = MakeNoPaddingConfig( in ChangeSpatialSizeOnSpaceToBatchedShape() local
1613 auto does_dim_have_padding = [](PaddingConfig padding_config, int64_t dim) { in SupportedOpForPropagation()
1830 PaddingConfig padding_config = in Propagate() local
2197 PaddingConfig padding_config = in Propagate() local
2787 PaddingConfig padding_config; in PropagateOnPad() local
2911 PaddingConfig padding_config = in PadAndSplitSpace() local
Dhlo_creation_utils.cc88 const PaddingConfig& padding_config, in MakePadHlo()
656 PaddingConfig padding_config; in PadVectorWithZeros() local
Dhlo.proto167 xla.PaddingConfig padding_config = 21; field
Dhlo_verifier_test.cc451 PaddingConfig padding_config; in TEST_F() local
475 PaddingConfig padding_config; in TEST_F() local
Dalgebraic_simplifier.cc1498 PaddingConfig padding_config; in HandleConcatenate() local
3711 PaddingConfig padding_config = pad->padding_config(); in HandlePad() local
5069 PaddingConfig padding_config; in HandleDynamicUpdateSlice() local
5717 PaddingConfig padding_config; in HandleReduceWindow() local
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc686 const PaddingConfig& padding_config) { in PadInternal()

12