| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| spacetobatch_op.cc | 53 xla::PaddingConfig padding_config; local
56 padding_config.add_dimensions(); // Don't pad the batch dimension.
58 auto* dim = padding_config.add_dimensions();
70 padding_config.add_dimensions();
77 b->Pad(input, XlaHelpers::Zero(b, input_dtype), padding_config);
|
| pooling_ops.cc | 477 xla::PaddingConfig padding_config = xla::MakeNoPaddingConfig(num_dims()); variable
480 auto* padding = padding_config.mutable_dimensions(dim);
488 ctx->builder()->Pad(out_backprop_div, zero, padding_config);
|
| strided_slice_op.cc | 177 xla::PaddingConfig padding_config; variable
180 auto* dims = padding_config.add_dimensions();
208 grad = ctx->builder()->Pad(grad, zero, padding_config);
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| pad_test.cc | 90 PaddingConfig padding_config; local
91 auto dimension = padding_config.add_dimensions();
97 AddParam(*Literal::CreateR0<float>(0.1), &b), padding_config);
105 PaddingConfig padding_config; local
106 auto dimension = padding_config.add_dimensions();
112 AddParam(*Literal::CreateR0<float>(0.1), &b), padding_config);
120 PaddingConfig padding_config; local
121 auto dimension = padding_config.add_dimensions();
127 AddParam(*Literal::CreateR0<float>(0.1), &b), padding_config);
186 PaddingConfig padding_config; local
228 PaddingConfig padding_config; local
325 PaddingConfig padding_config = MakeNoPaddingConfig(2); local
348 PaddingConfig padding_config = MakeNoPaddingConfig(2); local
373 PaddingConfig padding_config = MakeNoPaddingConfig(2); local
400 PaddingConfig padding_config = MakeNoPaddingConfig(2); local
428 PaddingConfig padding_config = MakeNoPaddingConfig(2); local
456 PaddingConfig padding_config = MakeNoPaddingConfig(3); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| pad_insertion.cc | 56 PaddingConfig padding_config = local
60 padding_config.mutable_dimensions(dim)->set_edge_padding_low(
62 padding_config.mutable_dimensions(dim)->set_edge_padding_high(
64 padding_config.mutable_dimensions(dim)->set_interior_padding(
75 padding_config)
77 input, padding, padding_config));
122 PaddingConfig padding_config; local
124 padding_config.add_dimensions();
128 padding_config.mutable_dimensions(dim)->set_interior_padding(
141 padding_config)
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| util.cc | 205 PaddingConfig padding_config; local
207 auto dimension = padding_config.add_dimensions();
212 return padding_config;
217 PaddingConfig padding_config; local
219 auto dimension = padding_config.add_dimensions();
224 return padding_config;
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_instruction_fusion_test.cc | 537 PaddingConfig padding_config; local
538 padding_config.add_dimensions()->set_edge_padding_high(4);
543 padding_config));
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_evaluator_test.cc | 459 PaddingConfig padding_config; local
462 auto dimension = padding_config.add_dimensions();
467 return padding_config;
481 auto padding_config = CreatePaddingConfig({{{1, 0, 2}}, {{0, 2, 1}}}); local
484 shape, operand_instruction, padding_value_instruction, padding_config));
590 PaddingConfig padding_config = MakeNoPaddingConfig(2); local
[all...] |
| hlo_instruction.h | 427 HloInstruction* padding_value, const PaddingConfig& padding_config);
1070 const PaddingConfig& padding_config() const { function in class:xla::HloInstruction
[all...] |
| shape_inference_test.cc | 1342 PaddingConfig padding_config; local
[all...] |
| algebraic_simplifier.cc | 469 PaddingConfig padding_config; local
471 auto padding_config_dim = padding_config.add_dimensions();
490 operands[pad_value_operand]->mutable_operand(0), padding_config));
[all...] |
