| /external/tensorflow/tensorflow/compiler/xla/service/ |
| transpose_folding_test.cc | 259 new_conv->convolution_dimension_numbers()
262 new_conv->convolution_dimension_numbers()
312 new_conv->convolution_dimension_numbers()
315 new_conv->convolution_dimension_numbers()
319 new_conv->convolution_dimension_numbers().kernel_spatial_dimensions(0));
322 new_conv->convolution_dimension_numbers().kernel_spatial_dimensions(1));
370 new_conv->convolution_dimension_numbers().input_batch_dimension());
373 new_conv->convolution_dimension_numbers().input_feature_dimension());
376 new_conv->convolution_dimension_numbers().input_spatial_dimensions(0));
379 new_conv->convolution_dimension_numbers().input_spatial_dimensions(1))
[all...] |
| hlo.proto | 74 xla.ConvolutionDimensionNumbers convolution_dimension_numbers = 16;
|
| transpose_folding.cc | 109 convolution.convolution_dimension_numbers();
|
| hlo_cost_analysis.cc | 383 const auto& dnums = convolution->convolution_dimension_numbers();
|
| hlo_instruction.h | 1078 const ConvolutionDimensionNumbers& convolution_dimension_numbers() const { function in class:xla::HloInstruction
[all...] |
| hlo_verifier.cc | 80 convolution->window(), convolution->convolution_dimension_numbers()));
[all...] |
| hlo_instruction.cc | 119 proto.convolution_dimension_numbers());
[all...] |
| algebraic_simplifier.cc | [all...] |
| hlo_evaluator.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| ir_emission_utils.cc | 52 convolution.convolution_dimension_numbers();
|
| conv_canonicalization.cc | 38 hlo->convolution_dimension_numbers();
|
| ir_emitter.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| cudnn_convolution_rewriter.cc | 39 conv->convolution_dimension_numbers();
80 conv->convolution_dimension_numbers();
226 ConvolutionDimensionNumbers dnums = conv->convolution_dimension_numbers();
388 conv->convolution_dimension_numbers().kernel_output_feature_dimension());
390 conv->convolution_dimension_numbers().kernel_input_feature_dimension());
426 conv->convolution_dimension_numbers());
|
| pad_insertion.cc | 155 conv->window(), conv->convolution_dimension_numbers(),
158 MaybePaddedKernel(conv->window(), conv->convolution_dimension_numbers(),
169 conv->convolution_dimension_numbers().kernel_spatial_dimensions(i)));
183 conv->convolution_dimension_numbers());
222 backward_conv->convolution_dimension_numbers();
285 backward_conv->convolution_dimension_numbers();
|
| cudnn_convolution_algorithm_picker.cc | 276 instr->window(), instr->convolution_dimension_numbers(), instr);
281 instr->convolution_dimension_numbers(), instr);
286 instr->window(), instr->convolution_dimension_numbers(), instr);
327 instr->convolution_dimension_numbers());
|
| gpu_layout_assignment.cc | 75 instr->convolution_dimension_numbers();
|
| ir_emitter_unnested.cc | 444 custom_call->window(), custom_call->convolution_dimension_numbers(),
457 custom_call->window(), custom_call->convolution_dimension_numbers(),
470 custom_call->window(), custom_call->convolution_dimension_numbers(),
[all...] |
