| /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 | 41 PaddingConfig MakeSymmetricPadding(absl::Span<const int64> sizes) { in MakeSymmetricPadding()
42 PaddingConfig config; in MakeSymmetricPadding()
146 bool HasSymmetricPadding(const PaddingConfig& padding_config) { in HasSymmetricPadding()
148 [](const PaddingConfig::PaddingConfigDimension& dim) { in HasSymmetricPadding()
|
| D | util.cc | 118 PaddingConfig MakeNoPaddingConfig(int64 rank) { in MakeNoPaddingConfig()
119 PaddingConfig padding_config; in MakeNoPaddingConfig()
129 PaddingConfig MakeEdgePaddingConfig( in MakeEdgePaddingConfig()
131 PaddingConfig padding_config; in MakeEdgePaddingConfig()
141 bool HasInteriorPadding(const PaddingConfig& config) { in HasInteriorPadding()
|
| D | window_util.h | 34 PaddingConfig MakeSymmetricPadding(absl::Span<const int64> sizes);
52 bool HasSymmetricPadding(const PaddingConfig& padding_config);
|
| D | util.h | 381 PaddingConfig MakeNoPaddingConfig(int64 rank);
385 PaddingConfig MakeEdgePaddingConfig(
390 bool HasInteriorPadding(const PaddingConfig& config);
|
| D | reference_util.h | 511 const Array2D<NativeT>& operand, const PaddingConfig& padding, in PadArray2D()
546 const PaddingConfig& padding, in PadArray3D()
604 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 | 66 StatusOr<PaddingConfig> ParsePaddingConfig(absl::string_view str);
|
| D | hlo_creation_utils.cc | 59 const PaddingConfig& padding_config) { in MakePadHlo()
425 PaddingConfig padding_config; in PadVectorWithZeros()
426 PaddingConfig::PaddingConfigDimension padding_config_dim; in PadVectorWithZeros()
|
| D | hlo_creation_utils.h | 46 const PaddingConfig& padding_config);
|
| D | hlo_verifier_test.cc | 304 PaddingConfig padding_config; in TEST_F()
328 PaddingConfig padding_config; in TEST_F()
|
| D | triangular_solve_expander.cc | 67 PaddingConfig padding_config = in DiagonalBlocks()
95 PaddingConfig config = MakeNoPaddingConfig(ndims); in DiagonalBlocks()
|
| D | hlo_instruction.h | 683 HloInstruction* padding_value, const PaddingConfig& padding_config);
1584 const PaddingConfig& padding_config() const;
1811 string PaddingConfigToString(const PaddingConfig& padding);
|
| D | shape_inference.h | 266 const PaddingConfig& padding_config);
|
| D | hlo_parser.cc | 90 StatusOr<PaddingConfig> ParsePaddingConfigOnly();
266 bool ParsePaddingConfig(PaddingConfig* padding);
1364 optional<PaddingConfig> padding; in ParseInstructionRhs()
2838 PaddingConfig result; in ParseAttributeHelper()
2842 static_cast<optional<PaddingConfig>*>(attr_out_ptr)->emplace(result); in ParseAttributeHelper()
3695 bool HloParser::ParsePaddingConfig(PaddingConfig* padding) { in ParsePaddingConfig()
4036 StatusOr<PaddingConfig> HloParser::ParsePaddingConfigOnly() { in ParsePaddingConfigOnly()
4038 PaddingConfig padding_config; in ParsePaddingConfigOnly()
4141 StatusOr<PaddingConfig> ParsePaddingConfig(absl::string_view str) { in ParsePaddingConfig()
|
| D | hlo_instructions.h | 1257 const PaddingConfig& padding_config);
1259 const PaddingConfig& padding_config() const { return padding_config_; } in padding_config()
1280 PaddingConfig padding_config_;
|
| D | algebraic_simplifier.cc | 750 PaddingConfig padding_config; in HandleConcatenate()
2204 PaddingConfig padding_config = pad->padding_config(); in HandlePad()
2251 PaddingConfig nonzero_padding = pad->padding_config(); in HandlePad()
2253 PaddingConfig::PaddingConfigDimension* padding_dimension = in HandlePad()
2277 const PaddingConfig::PaddingConfigDimension& padding_dimension = in HandlePad()
3180 const PaddingConfig& pad_config = pad->padding_config(); in HandleReduceWindow()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_padding_legalization.cc | 62 PaddingConfig padding_config = in MaybePaddedAndSlicedInput()
131 PaddingConfig padding_config; in MaybePaddedKernel()
221 PaddingConfig input_padding_config = in CanonicalizeBackwardFilterConvolution()
|
| D | cudnn_conv_pad_for_tensor_cores.cc | 53 PaddingConfig pad_config = MakeNoPaddingConfig(shape.rank()); in PadInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_data.i | 479 %typemap(in) const PaddingConfig&
480 (PaddingConfig padding_config) {
508 PaddingConfig::PaddingConfigDimension* dimension =
|
| D | local_computation_builder.h | 248 const PaddingConfig& padding_config);
|
| /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 | spacetobatch_op.cc | 53 xla::PaddingConfig padding_config; in SpaceToBatch()
|
| D | strided_slice_op.cc | 179 xla::PaddingConfig padding_config; in Compile()
|