| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | pad_test.cc | 69 PaddingConfig r4_padding_on_dim0_dim1_;
90 PaddingConfig padding_config; in XLA_TEST_P()
105 PaddingConfig padding_config; in XLA_TEST_P()
120 PaddingConfig padding_config; in XLA_TEST_P()
188 PaddingConfig padding_config; in TEST_P()
230 PaddingConfig padding_config; in XLA_TEST_P()
327 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P()
349 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P()
373 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P()
400 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | window_util.cc | 55 PaddingConfig MakeSymmetricPadding(absl::Span<const int64> sizes) { in MakeSymmetricPadding()
56 PaddingConfig config; in MakeSymmetricPadding()
162 bool HasSymmetricPadding(const PaddingConfig& padding_config) { in HasSymmetricPadding()
164 [](const PaddingConfig::PaddingConfigDimension& dim) { in HasSymmetricPadding()
|
| D | window_util.h | 38 PaddingConfig MakeSymmetricPadding(absl::Span<const int64> sizes);
56 bool HasSymmetricPadding(const PaddingConfig& padding_config);
|
| D | util.cc | 133 PaddingConfig MakeNoPaddingConfig(int64 rank) { in MakeNoPaddingConfig()
134 PaddingConfig padding_config; in MakeNoPaddingConfig()
144 PaddingConfig MakeEdgePaddingConfig( in MakeEdgePaddingConfig()
146 PaddingConfig padding_config; in MakeEdgePaddingConfig()
156 bool HasInteriorPadding(const PaddingConfig& config) { in HasInteriorPadding()
|
| D | util.h | 389 PaddingConfig MakeNoPaddingConfig(int64 rank);
393 PaddingConfig MakeEdgePaddingConfig(
398 bool HasInteriorPadding(const PaddingConfig& config);
|
| D | reference_util.h | 503 const Array2D<NativeT>& operand, const PaddingConfig& padding, in PadArray2D()
538 const PaddingConfig& padding, in PadArray3D()
596 const PaddingConfig& padding, in PadArray4D()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling.cc | 54 PaddingConfig padding_config; in AvgPoolDivideByCountWithGeneralPadding()
89 PaddingConfig MakeSpatialPaddingConfig( in MakeSpatialPaddingConfig()
93 PaddingConfig padding_config; in MakeSpatialPaddingConfig()
242 PaddingConfig padding_config = MakeNoPaddingConfig(num_dims); in AvgPoolGrad()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 457 class PaddingConfig(object): class
466 padding_config: Union[PaddingConfig, Sequence[Tuple[int, int, int]]] argument
467 ) -> PaddingConfig:
480 padding_config = PaddingConfig()
|
| D | types.h | 364 struct type_caster<xla::PaddingConfig> {
366 PYBIND11_TYPE_CASTER(xla::PaddingConfig, _("xla::PaddingConfig"));
374 xla::PaddingConfig::PaddingConfigDimension* config_dim =
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_parser.h | 75 StatusOr<PaddingConfig> ParsePaddingConfig(absl::string_view str);
|
| D | hlo_creation_utils.cc | 70 const PaddingConfig& padding_config) { in MakePadHlo()
537 PaddingConfig padding_config; in PadVectorWithZeros()
538 PaddingConfig::PaddingConfigDimension padding_config_dim; in PadVectorWithZeros()
|
| D | hlo_creation_utils.h | 51 const PaddingConfig& padding_config);
|
| D | hlo_instruction.h | 892 HloInstruction* padding_value, const PaddingConfig& padding_config);
1870 const PaddingConfig& padding_config() const;
1871 PaddingConfig* mutable_padding_config();
2162 string PaddingConfigToString(const PaddingConfig& padding);
|
| D | triangular_solve_expander.cc | 70 PaddingConfig padding_config = in DiagonalBlocks()
98 PaddingConfig config = MakeNoPaddingConfig(ndims); in DiagonalBlocks()
|
| D | shape_inference.h | 295 const PaddingConfig& padding_config);
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_padding_legalization.cc | 62 PaddingConfig padding_config = in MaybePaddedAndSlicedInput()
131 PaddingConfig padding_config; in MaybePaddedKernel()
221 PaddingConfig input_padding_config = in CanonicalizeBackwardFilterConvolution()
|
| D | cublas_gemm_pad_for_tensor_cores.cc | 68 PaddingConfig padding_config; in PadForTensorCores()
|
| D | tree_reduction_rewriter.cc | 125 PaddingConfig padding_config = MakeNoPaddingConfig(input_shape.rank()); in RewriteReduction()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pad_op.cc | 58 xla::PaddingConfig config; in Compile()
|
| D | xla_pad_op.cc | 72 xla::PaddingConfig padding_config; in Compile()
|
| D | slice_op.cc | 153 xla::PaddingConfig padding_config; in Compile()
|
| D | spacetobatch_op.cc | 53 xla::PaddingConfig padding_config; in SpaceToBatch()
|
| D | matrix_diag_ops.cc | 126 xla::PaddingConfig padding_config; in SetMatrixDiag()
400 xla::PaddingConfig padding_config = in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | mlir_hlo_builder_test.cc | 161 PaddingConfig padding_config; in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.h | 256 const PaddingConfig& padding_config) override;
|