| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transpose_folding_test.cc | 267 new_conv->convolution_dimension_numbers() in TEST_F()
270 new_conv->convolution_dimension_numbers() in TEST_F()
323 new_conv->convolution_dimension_numbers() in TEST_F()
326 new_conv->convolution_dimension_numbers() in TEST_F()
330 new_conv->convolution_dimension_numbers().kernel_spatial_dimensions(0)); in TEST_F()
333 new_conv->convolution_dimension_numbers().kernel_spatial_dimensions(1)); in TEST_F()
384 new_conv->convolution_dimension_numbers().input_batch_dimension()); in TEST_F()
387 new_conv->convolution_dimension_numbers().input_feature_dimension()); in TEST_F()
390 new_conv->convolution_dimension_numbers().input_spatial_dimensions(0)); in TEST_F()
393 new_conv->convolution_dimension_numbers().input_spatial_dimensions(1)); in TEST_F()
[all …]
|
| D | dynamic_padder.cc | 98 inst->convolution_dimension_numbers().input_batch_dimension() == in ShouldSkipPadOnOperand()
|
| D | convolution_group_converter.cc | 202 auto dim_numbers = convolution->convolution_dimension_numbers(); in HandleBatchGroupCount()
334 auto dim_numbers = convolution->convolution_dimension_numbers(); in HandleConvolution()
|
| D | transpose_folding.cc | 128 convolution.convolution_dimension_numbers(); in FoldTransposeIntoConvolution()
|
| D | dynamic_dimension_inference.cc | 268 conv->convolution_dimension_numbers(); in HandleConvolution()
|
| D | hlo_instructions.cc | 1935 convolution_dimension_numbers(), in IdenticalSlowPath()
1936 casted_other.convolution_dimension_numbers()) && in IdenticalSlowPath()
2150 convolution_dimension_numbers(), in IdenticalSlowPath()
2151 casted_other.convolution_dimension_numbers()))) { in IdenticalSlowPath()
|
| D | hlo_instruction.cc | 425 proto.convolution_dimension_numbers(), precision_config); in CreateFromProto()
468 proto.convolution_dimension_numbers()); in CreateFromProto()
3439 HloInstruction::convolution_dimension_numbers() const { in convolution_dimension_numbers() function in xla::HloInstruction
3441 return convolution->convolution_dimension_numbers(); in convolution_dimension_numbers()
3444 return custom_call->convolution_dimension_numbers(); in convolution_dimension_numbers()
|
| D | hlo_instructions.h | 1038 const ConvolutionDimensionNumbers& convolution_dimension_numbers() const { in convolution_dimension_numbers() function
1191 const ConvolutionDimensionNumbers& convolution_dimension_numbers() const { in convolution_dimension_numbers() function
|
| D | hlo_cost_analysis.cc | 442 const auto& dnums = convolution->convolution_dimension_numbers(); in HandleConvolution()
|
| D | hlo.proto | 79 xla.ConvolutionDimensionNumbers convolution_dimension_numbers = 16; field
|
| D | hlo_instruction_test.cc | 1740 clone->convolution_dimension_numbers(), dnums)) in TEST_F()
1741 << clone->convolution_dimension_numbers().DebugString(); in TEST_F()
|
| D | hlo_instruction.h | 1549 const ConvolutionDimensionNumbers& convolution_dimension_numbers() const;
|
| D | algebraic_simplifier.cc | 3399 convolution->convolution_dimension_numbers(); in FoldConvInputPad()
3465 convolution->convolution_dimension_numbers(); in FoldConvFilterPad()
3532 convolution->convolution_dimension_numbers(); in SimplifyConvToDot()
|
| D | hlo_verifier.cc | 160 convolution->window(), convolution->convolution_dimension_numbers())); in HandleConvolution()
|
| D | hlo_evaluator_typed_visitor.h | 1024 const auto& dnums = conv->convolution_dimension_numbers(); in HandleConvolution()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_rewriter.cc | 69 conv->convolution_dimension_numbers(); in CanImplementAsCudnnForwardConv()
117 conv->convolution_dimension_numbers(); in MatchBackwardFilter()
273 ConvolutionDimensionNumbers dnums = conv->convolution_dimension_numbers(); in MatchBackwardInput()
434 conv->convolution_dimension_numbers().kernel_output_feature_dimension()); in MatchBackwardInput()
436 conv->convolution_dimension_numbers().kernel_input_feature_dimension()); in MatchBackwardInput()
527 conv->convolution_dimension_numbers(), in RunOnInstruction()
|
| D | cudnn_conv_padding_legalization.cc | 159 &new_conv_window, conv->convolution_dimension_numbers(), in CanonicalizeForwardConvolution()
162 MaybePaddedKernel(new_conv_window, conv->convolution_dimension_numbers(), in CanonicalizeForwardConvolution()
172 conv->convolution_dimension_numbers().kernel_spatial_dimensions(i))); in CanonicalizeForwardConvolution()
224 backward_conv->convolution_dimension_numbers(); in CanonicalizeBackwardFilterConvolution()
283 backward_conv->convolution_dimension_numbers(); in CanonicalizeBackwardInputConvolution()
|
| D | cudnn_fused_conv_rewriter.cc | 158 conv->convolution_dimension_numbers().output_feature_dimension()) { in FindConvWithRelu()
211 conv->convolution_dimension_numbers().output_feature_dimension()); in TryRewriteToCudnnForwardRelu()
228 conv->convolution_dimension_numbers()); in TryRewriteToCudnnForwardRelu()
|
| D | cudnn_conv_pad_for_tensor_cores.cc | 123 const auto& dnums = conv->convolution_dimension_numbers(); in PadForTensorCores()
|
| D | gpu_layout_assignment.cc | 141 instr->convolution_dimension_numbers(), input, filter, output)); in AddBackendConstraintsToDnnConvCustomCall()
|
| D | cudnn_conv_runner.cc | 335 params.dnums = &conv->convolution_dimension_numbers(); in GetCudnnConvParams()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emission_utils.cc | 84 convolution.convolution_dimension_numbers(); in PotentiallyImplementedAsEigenConvolution()
|
| D | conv_canonicalization.cc | 39 hlo->convolution_dimension_numbers(); in Run()
|
| D | ir_emitter.cc | 960 convolution->convolution_dimension_numbers(); in EmitElementalConvolution()
1121 convolution->convolution_dimension_numbers(); in HandleConvolution()
|