| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_gather_to_torch_index_select.cc | 50 auto dimension_numbers = gather.dimension_numbers(); in matchAndRewrite() local
|
| D | optimize_mhlo.cc | 59 auto dimension_numbers = gather.dimension_numbers(); in matchAndRewrite() local
|
| D | hlo_legalize_to_lhlo.cc | 161 auto dimension_numbers = mhlo::DotDimensionNumbers::get( in matchAndRewrite() local
|
| D | legalize_to_linalg.cc | 272 if (const mhlo::ConvDimensionNumbers& dimension_numbers = in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 249 dimension_numbers, argument
297 def dot_general(lhs, rhs, dimension_numbers, precision_config=None, name=None): argument
482 def gather(operand, start_indices, dimension_numbers, slice_sizes, argument
494 dimension_numbers, indices_are_sorted=False, name=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_dimension_numbers_test.cc | 43 ConvolutionDimensionNumbers dimension_numbers; in CreateConvDimensionNumbers() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cublas_gemm_pad_for_tensor_cores.cc | 122 auto dimension_numbers = dot->dot_dimension_numbers(); in CheckCanonical() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1291 DotDimensionNumbers dimension_numbers; in Dot() local
1300 XlaOp lhs, XlaOp rhs, const DotDimensionNumbers& dimension_numbers, in DotGeneral()
1317 const DotDimensionNumbers& dimension_numbers, in DotGeneralInternal()
1399 Padding padding, const ConvolutionDimensionNumbers& dimension_numbers, in ConvWithGeneralDimensions()
1438 const ConvolutionDimensionNumbers& dimension_numbers, in ConvGeneral()
1452 const ConvolutionDimensionNumbers& dimension_numbers, in ConvGeneralDilated()
1490 const ConvolutionDimensionNumbers& dimension_numbers, in DynamicConvInstruction()
1532 const ConvolutionDimensionNumbers& dimension_numbers, in DynamicConvInputGrad()
1555 const ConvolutionDimensionNumbers& dimension_numbers, in DynamicConvKernelGrad()
1581 const ConvolutionDimensionNumbers& dimension_numbers, in DynamicConvForward()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 98 const ConvolutionDimensionNumbers& dimension_numbers, in ConvGeneralDilatedInternal()
289 const GatherDimensionNumbers& dimension_numbers, in GatherInternal()
303 const ScatterDimensionNumbers& dimension_numbers, bool indices_are_sorted, in ScatterInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util_test.cc | 333 ConvolutionDimensionNumbers dimension_numbers; in TEST_F() local
389 ConvolutionDimensionNumbers dimension_numbers; in TEST_F() local
|
| D | reference_util.cc | 402 ConvolutionDimensionNumbers dimension_numbers) { in ConvArray4DGeneralDimensions()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_resize_ops.cc | 246 xla::ConvolutionDimensionNumbers dimension_numbers; in ResizeUsingDilationAndConvolution() local
387 xla::ConvolutionDimensionNumbers dimension_numbers; in ResizeUsingDilationAndConvolutionGradOp() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 183 xla::DotDimensionNumbers dimension_numbers; in __anonbbc1c2d10202() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference.cc | 478 const DotDimensionNumbers& dimension_numbers = in HandleDot() local
566 const ConvolutionDimensionNumbers& dimension_numbers = in HandleConvolution() local
694 const ConvolutionDimensionNumbers& dimension_numbers = in HandleDynamicConvolutionForward() local
|
| D | hlo_creation_utils.cc | 96 const ConvolutionDimensionNumbers& dimension_numbers, in MakeConvolveHlo()
|
| D | shape_inference.cc | 599 const DotDimensionNumbers& dimension_numbers) { in ValidateDotDimensionNumbers()
650 const DotDimensionNumbers& dimension_numbers, in InferDotOpShape()
|
| D | hlo_instruction.cc | 986 const ConvolutionDimensionNumbers& dimension_numbers, in CreateConvolve()
1028 const DotDimensionNumbers& dimension_numbers, in CreateDot()
|
| D | hlo_instructions.cc | 2136 const ConvolutionDimensionNumbers& dimension_numbers, in HloConvolutionInstruction()
2949 const DotDimensionNumbers& dimension_numbers, in HloDotInstruction()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 947 xla::ScatterDimensionNumbers dimension_numbers = in ExportXlaOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 550 auto dimension_numbers = DotDimensionNumbers::get( in BatchDot() local
2857 auto dimension_numbers = DotDimensionNumbers::get( in matchAndRewrite() local
|