| /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 | 169 PaddingConfig MakeNoPaddingConfig(int64 rank) { in MakeNoPaddingConfig()
170 PaddingConfig padding_config; in MakeNoPaddingConfig()
180 PaddingConfig MakeEdgePaddingConfig( in MakeEdgePaddingConfig()
182 PaddingConfig padding_config; in MakeEdgePaddingConfig()
192 bool HasInteriorPadding(const PaddingConfig& config) { in HasInteriorPadding()
|
| D | util.h | 421 PaddingConfig MakeNoPaddingConfig(int64 rank);
425 PaddingConfig MakeEdgePaddingConfig(
430 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()
239 PaddingConfig padding_config = MakeNoPaddingConfig(num_dims); in AvgPoolGrad()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_parser.h | 75 StatusOr<PaddingConfig> ParsePaddingConfig(absl::string_view str);
|
| D | hlo_creation_utils.cc | 60 const PaddingConfig& padding_config) { in MakePadHlo()
478 PaddingConfig padding_config; in PadVectorWithZeros()
479 PaddingConfig::PaddingConfigDimension padding_config_dim; in PadVectorWithZeros()
|
| D | hlo_creation_utils.h | 46 const PaddingConfig& padding_config);
|
| D | triangular_solve_expander.cc | 67 PaddingConfig padding_config = in DiagonalBlocks()
95 PaddingConfig config = MakeNoPaddingConfig(ndims); in DiagonalBlocks()
|
| D | shape_inference.h | 271 const PaddingConfig& padding_config);
|
| D | hlo_parser.cc | 96 StatusOr<PaddingConfig> ParsePaddingConfigOnly();
277 bool ParsePaddingConfig(PaddingConfig* padding);
1426 optional<PaddingConfig> padding; in ParseInstructionRhs()
2921 PaddingConfig result; in ParseAttributeHelper()
2925 static_cast<optional<PaddingConfig>*>(attr_out_ptr)->emplace(result); in ParseAttributeHelper()
3835 bool HloParserImpl::ParsePaddingConfig(PaddingConfig* padding) { in ParsePaddingConfig()
4203 StatusOr<PaddingConfig> HloParserImpl::ParsePaddingConfigOnly() { in ParsePaddingConfigOnly()
4205 PaddingConfig padding_config; in ParsePaddingConfigOnly()
4319 StatusOr<PaddingConfig> ParsePaddingConfig(absl::string_view str) { in ParsePaddingConfig()
|
| D | hlo_verifier_test.cc | 358 PaddingConfig padding_config; in TEST_F()
382 PaddingConfig padding_config; in TEST_F()
|
| D | hlo_instruction.h | 829 HloInstruction* padding_value, const PaddingConfig& padding_config);
1762 const PaddingConfig& padding_config() const;
2024 string PaddingConfigToString(const PaddingConfig& padding);
|
| D | hlo_instructions.h | 1344 const PaddingConfig& padding_config);
1346 const PaddingConfig& padding_config() const { return padding_config_; } in padding_config()
1367 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 | 111 PaddingConfig padding_config = MakeNoPaddingConfig(input_shape.rank()); in RewriteReduction()
|
| D | cudnn_pad_for_convolutions.cc | 53 PaddingConfig pad_config = MakeNoPaddingConfig(shape.rank()); in PadInstruction()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pad_op.cc | 58 xla::PaddingConfig config; in Compile()
|
| D | xla_pad_op.cc | 80 xla::PaddingConfig padding_config; in Compile()
|
| D | matrix_diag_ops.cc | 126 xla::PaddingConfig padding_config; in SetMatrixDiag()
399 xla::PaddingConfig padding_config = in Compile()
|
| D | spacetobatch_op.cc | 53 xla::PaddingConfig padding_config; in SpaceToBatch()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | types.h | 408 struct type_caster<xla::PaddingConfig> {
410 PYBIND11_TYPE_CASTER(xla::PaddingConfig, _("xla::PaddingConfig"));
418 xla::PaddingConfig::PaddingConfigDimension* config_dim =
|