| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | pad_test.cc | 90 PaddingConfig padding_config; in XLA_TEST_P() local
105 PaddingConfig padding_config; in XLA_TEST_P() local
120 PaddingConfig padding_config; in XLA_TEST_P() local
188 PaddingConfig padding_config; in TEST_P() local
230 PaddingConfig padding_config; in XLA_TEST_P() local
327 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
349 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
373 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
400 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
428 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling.cc | 54 PaddingConfig padding_config; in AvgPoolDivideByCountWithGeneralPadding() local
93 PaddingConfig padding_config; in MakeSpatialPaddingConfig() local
161 auto padding_config = MakeSpatialPaddingConfig(padding, num_spatial_dims, in AvgPool() local
239 PaddingConfig padding_config = MakeNoPaddingConfig(num_dims); in AvgPoolGrad() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_pad_op.cc | 80 xla::PaddingConfig padding_config; in Compile() local
|
| D | spacetobatch_op.cc | 53 xla::PaddingConfig padding_config; in SpaceToBatch() local
|
| D | strided_slice_op.cc | 179 xla::PaddingConfig padding_config; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | util.cc | 119 PaddingConfig padding_config; in MakeNoPaddingConfig() local
131 PaddingConfig padding_config; in MakeEdgePaddingConfig() local
|
| D | window_util.cc | 146 bool HasSymmetricPadding(const PaddingConfig& padding_config) { in HasSymmetricPadding()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_padding_legalization.cc | 62 PaddingConfig padding_config = in MaybePaddedAndSlicedInput() local
131 PaddingConfig padding_config; in MaybePaddedKernel() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 59 const PaddingConfig& padding_config) { in MakePadHlo()
425 PaddingConfig padding_config; in PadVectorWithZeros() local
|
| D | hlo_verifier_test.cc | 304 PaddingConfig padding_config; in TEST_F() local
328 PaddingConfig padding_config; in TEST_F() local
|
| D | triangular_solve_expander.cc | 67 PaddingConfig padding_config = in DiagonalBlocks() local
|
| D | hlo.proto | 106 xla.PaddingConfig padding_config = 21; field
|
| D | hlo_instructions.h | 1259 const PaddingConfig& padding_config() const { return padding_config_; } in padding_config() function
|
| D | algebraic_simplifier.cc | 750 PaddingConfig padding_config; in HandleConcatenate() local
2204 PaddingConfig padding_config = pad->padding_config(); in HandlePad() local
2683 auto padding_config = pad->padding_config(); in TrySimplifyScalarSlice() local
|
| D | dynamic_dimension_inference.cc | 151 const PaddingConfig_PaddingConfigDimension& padding_config = in HandlePad() local
|
| D | hlo_evaluator_test.cc | 560 PaddingConfig padding_config; in CreatePaddingConfig() local
582 auto padding_config = CreatePaddingConfig({{{1, 0, 2}}, {{0, 2, 1}}}); in TEST_F() local
691 PaddingConfig padding_config = MakeNoPaddingConfig(2); in TEST_P() local
|
| D | hlo_instruction.cc | 1169 const PaddingConfig& padding_config) { in CreatePad()
3504 const PaddingConfig& HloInstruction::padding_config() const { in padding_config() function in xla::HloInstruction
|
| D | shape_inference.cc | 486 const PaddingConfig& padding_config) { in InferPadShape()
|
| D | hlo_parser.cc | 4038 PaddingConfig padding_config; in ParsePaddingConfigOnly() local
|
| D | shape_inference_test.cc | 1423 PaddingConfig padding_config; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 483 const PaddingConfig& padding_config) { in Pad()
|
| D | xla_client.py | 1171 def Pad(self, operand, padding_value, padding_config): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 651 PaddingConfig padding_config; in TEST_F() local
|
| D | xla_builder.cc | 872 const PaddingConfig& padding_config) { in Pad()
2851 const PaddingConfig& padding_config) { in Pad()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2128 const PaddingConfig& padding_config = pad->padding_config(); in HandlePad() local
|