Home
last modified time | relevance | path

Searched refs:padding_config (Results 1 – 25 of 39) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dpad_test.cc90 PaddingConfig padding_config; in XLA_TEST_P() local
91 auto dimension = padding_config.add_dimensions(); in XLA_TEST_P()
97 AddParam(LiteralUtil::CreateR0<float>(0.1), &b), padding_config); in XLA_TEST_P()
105 PaddingConfig padding_config; in XLA_TEST_P() local
106 auto dimension = padding_config.add_dimensions(); in XLA_TEST_P()
112 AddParam(LiteralUtil::CreateR0<float>(0.1), &b), padding_config); in XLA_TEST_P()
120 PaddingConfig padding_config; in XLA_TEST_P() local
121 auto dimension = padding_config.add_dimensions(); in XLA_TEST_P()
127 AddParam(LiteralUtil::CreateR0<float>(0.1), &b), padding_config); in XLA_TEST_P()
188 PaddingConfig padding_config; in TEST_P() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc54 PaddingConfig padding_config; in AvgPoolDivideByCountWithGeneralPadding() local
56 auto dims = padding_config.add_dimensions(); in AvgPoolDivideByCountWithGeneralPadding()
61 auto padded_ones = Pad(ones, zero, padding_config); in AvgPoolDivideByCountWithGeneralPadding()
93 PaddingConfig padding_config; in MakeSpatialPaddingConfig() local
95 padding_config.add_dimensions(); in MakeSpatialPaddingConfig()
101 auto padding_dimension = padding_config.mutable_dimensions(dim); in MakeSpatialPaddingConfig()
105 return padding_config; in MakeSpatialPaddingConfig()
161 auto padding_config = MakeSpatialPaddingConfig(padding, num_spatial_dims, in AvgPool() local
163 auto padded_operand = Pad(operand, Zero(b, dtype), padding_config); in AvgPool()
239 PaddingConfig padding_config = MakeNoPaddingConfig(num_dims); in AvgPoolGrad() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_padding_legalization.cc62 PaddingConfig padding_config = in MaybePaddedAndSlicedInput() local
67 padding_config.mutable_dimensions(dim)->set_edge_padding_low( in MaybePaddedAndSlicedInput()
72 padding_config.mutable_dimensions(dim)->set_edge_padding_high( in MaybePaddedAndSlicedInput()
77 padding_config.mutable_dimensions(dim)->set_interior_padding( in MaybePaddedAndSlicedInput()
85 input = MakePadHlo(input, padding, padding_config).ValueOrDie(); in MaybePaddedAndSlicedInput()
131 PaddingConfig padding_config; in MaybePaddedKernel() local
133 padding_config.add_dimensions(); in MaybePaddedKernel()
137 padding_config.mutable_dimensions(dim)->set_interior_padding( in MaybePaddedKernel()
145 return MakePadHlo(kernel, padding, padding_config).ValueOrDie(); in MaybePaddedKernel()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dspacetobatch_op.cc53 xla::PaddingConfig padding_config; in SpaceToBatch() local
56 padding_config.add_dimensions(); // Don't pad the batch dimension. in SpaceToBatch()
58 auto* dim = padding_config.add_dimensions(); in SpaceToBatch()
70 padding_config.add_dimensions(); in SpaceToBatch()
77 xla::Pad(input, XlaHelpers::Zero(b, input_dtype), padding_config); in SpaceToBatch()
Dxla_pad_op.cc80 xla::PaddingConfig padding_config; in Compile() local
82 auto* dim = padding_config.add_dimensions(); in Compile()
90 padding_config); in Compile()
Dstrided_slice_op.cc179 xla::PaddingConfig padding_config; in Compile() local
182 auto* dims = padding_config.add_dimensions(); in Compile()
210 grad = xla::Pad(grad, zero, padding_config); in Compile()
/external/tensorflow/tensorflow/compiler/xla/
Dutil.cc119 PaddingConfig padding_config; in MakeNoPaddingConfig() local
121 auto dimension = padding_config.add_dimensions(); in MakeNoPaddingConfig()
126 return padding_config; in MakeNoPaddingConfig()
131 PaddingConfig padding_config; in MakeEdgePaddingConfig() local
133 auto dimension = padding_config.add_dimensions(); in MakeEdgePaddingConfig()
138 return padding_config; in MakeEdgePaddingConfig()
Dwindow_util.cc146 bool HasSymmetricPadding(const PaddingConfig& padding_config) { in HasSymmetricPadding() argument
147 return absl::c_all_of(padding_config.dimensions(), in HasSymmetricPadding()
Dwindow_util.h52 bool HasSymmetricPadding(const PaddingConfig& padding_config);
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier_test.cc304 PaddingConfig padding_config; in TEST_F() local
305 padding_config.add_dimensions()->set_interior_padding(-1); in TEST_F()
310 padding_config)); in TEST_F()
328 PaddingConfig padding_config; in TEST_F() local
329 padding_config.add_dimensions()->set_interior_padding(-1); in TEST_F()
334 padding_config)); in TEST_F()
Dhlo_creation_utils.cc59 const PaddingConfig& padding_config) { in MakePadHlo() argument
65 padding_config)); in MakePadHlo()
67 pad_shape, operand, padding_value, padding_config)); in MakePadHlo()
425 PaddingConfig padding_config; in PadVectorWithZeros() local
429 *padding_config.add_dimensions() = padding_config_dim; in PadVectorWithZeros()
434 return MakePadHlo(operand, zero, padding_config); in PadVectorWithZeros()
Ddynamic_dimension_inference.cc151 const PaddingConfig_PaddingConfigDimension& padding_config = in HandlePad() local
152 hlo->padding_config().dimensions(dimension); in HandlePad()
153 if (padding_config.interior_padding() == 0 && in HandlePad()
154 padding_config.edge_padding_low() == 0 && in HandlePad()
155 padding_config.edge_padding_high() == 0) { in HandlePad()
Dalgebraic_simplifier.cc750 PaddingConfig padding_config; in HandleConcatenate() local
752 auto padding_config_dim = padding_config.add_dimensions(); in HandleConcatenate()
771 operands[pad_value_operand]->mutable_operand(0), padding_config)); in HandleConcatenate()
2203 if (HasInteriorPadding(pad->padding_config())) { in HandlePad()
2204 PaddingConfig padding_config = pad->padding_config(); in HandlePad() local
2207 if (padding_config.dimensions(i).interior_padding() > 0 && in HandlePad()
2210 padding_config.mutable_dimensions(i)->set_interior_padding(0); in HandlePad()
2217 pad->mutable_operand(1), padding_config)); in HandlePad()
2225 for (auto& padding_dimension : pad->padding_config().dimensions()) { in HandlePad()
2251 PaddingConfig nonzero_padding = pad->padding_config(); in HandlePad()
[all …]
Dhlo_creation_utils.h46 const PaddingConfig& padding_config);
Dtriangular_solve_expander.cc67 PaddingConfig padding_config = in DiagonalBlocks() local
70 Pad(start_indices, ConstantR0<int32>(builder, 0), padding_config); in DiagonalBlocks()
Dshape_inference.h266 const PaddingConfig& padding_config);
Dshape_inference.cc486 const PaddingConfig& padding_config) { in InferPadShape() argument
495 if (operand_shape.rank() != padding_config.dimensions_size()) { in InferPadShape()
500 padding_config.ShortDebugString()); in InferPadShape()
507 if (absl::c_any_of(padding_config.dimensions(), in InferPadShape()
512 padding_config.ShortDebugString()); in InferPadShape()
522 const auto& p = padding_config.dimensions(i); in InferPadShape()
Dhlo_evaluator_test.cc560 PaddingConfig padding_config; in CreatePaddingConfig() local
563 auto dimension = padding_config.add_dimensions(); in CreatePaddingConfig()
568 return padding_config; in CreatePaddingConfig()
582 auto padding_config = CreatePaddingConfig({{{1, 0, 2}}, {{0, 2, 1}}}); in TEST_F() local
585 shape, operand_instruction, padding_value_instruction, padding_config)); in TEST_F()
691 PaddingConfig padding_config = MakeNoPaddingConfig(2); in TEST_P() local
Dhlo_instruction.h683 HloInstruction* padding_value, const PaddingConfig& padding_config);
1584 const PaddingConfig& padding_config() const;
Dshape_inference_test.cc1423 PaddingConfig padding_config; in TEST_F() local
1424 auto dimension0 = padding_config.add_dimensions(); in TEST_F()
1428 auto dimension1 = padding_config.add_dimensions(); in TEST_F()
1434 input_shape, padding_value_shape, padding_config); in TEST_F()
1443 input_shape, padding_value_shape, padding_config); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py1171 def Pad(self, operand, padding_value, padding_config): argument
1184 if isinstance(padding_config, tuple) or isinstance(padding_config, list):
1185 padding_config = GetPaddingConfigFromTriples(padding_config)
1186 return self._client.Pad(operand, padding_value, padding_config)
1897 padding_config = PaddingConfig()
1903 padding_config.dimensions.append(dimension)
1904 return padding_config
Dxla_data.i480 (PaddingConfig padding_config) {
509 padding_config.add_dimensions();
516 $1 = &padding_config;
Dlocal_computation_builder.cc483 const PaddingConfig& padding_config) { in Pad() argument
484 return xla::Pad(operand.op(), padding_value.op(), padding_config); in Pad()
Dlocal_computation_builder.h248 const PaddingConfig& padding_config);
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc651 PaddingConfig padding_config; in TEST_F() local
653 auto dimension = padding_config.add_dimensions(); in TEST_F()
658 Pad(gte, pad_val, padding_config); in TEST_F()

12