/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | pad_test.cc | 90 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/ |
D | pooling.cc | 54 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/ |
D | cudnn_conv_padding_legalization.cc | 62 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/ |
D | spacetobatch_op.cc | 53 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()
|
D | xla_pad_op.cc | 80 xla::PaddingConfig padding_config; in Compile() local 82 auto* dim = padding_config.add_dimensions(); in Compile() 90 padding_config); in Compile()
|
D | strided_slice_op.cc | 179 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/ |
D | util.cc | 119 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()
|
D | window_util.cc | 146 bool HasSymmetricPadding(const PaddingConfig& padding_config) { in HasSymmetricPadding() argument 147 return absl::c_all_of(padding_config.dimensions(), in HasSymmetricPadding()
|
D | window_util.h | 52 bool HasSymmetricPadding(const PaddingConfig& padding_config);
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_verifier_test.cc | 304 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()
|
D | hlo_creation_utils.cc | 59 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()
|
D | dynamic_dimension_inference.cc | 151 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()
|
D | algebraic_simplifier.cc | 750 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 …]
|
D | hlo_creation_utils.h | 46 const PaddingConfig& padding_config);
|
D | triangular_solve_expander.cc | 67 PaddingConfig padding_config = in DiagonalBlocks() local 70 Pad(start_indices, ConstantR0<int32>(builder, 0), padding_config); in DiagonalBlocks()
|
D | shape_inference.h | 266 const PaddingConfig& padding_config);
|
D | shape_inference.cc | 486 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()
|
D | hlo_evaluator_test.cc | 560 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
|
D | hlo_instruction.h | 683 HloInstruction* padding_value, const PaddingConfig& padding_config); 1584 const PaddingConfig& padding_config() const;
|
D | shape_inference_test.cc | 1423 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/ |
D | xla_client.py | 1171 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
|
D | xla_data.i | 480 (PaddingConfig padding_config) { 509 padding_config.add_dimensions(); 516 $1 = &padding_config;
|
D | local_computation_builder.cc | 483 const PaddingConfig& padding_config) { in Pad() argument 484 return xla::Pad(operand.op(), padding_value.op(), padding_config); in Pad()
|
D | local_computation_builder.h | 248 const PaddingConfig& padding_config);
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder_test.cc | 651 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()
|