| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.cc | 1364 DotDimensionNumbers dimension_numbers; in Dot() local
1374 XlaOp lhs, XlaOp rhs, const DotDimensionNumbers& dimension_numbers, in DotGeneral()
1391 const DotDimensionNumbers& dimension_numbers, in DotGeneralInternal()
1472 Padding padding, const ConvolutionDimensionNumbers& dimension_numbers, in ConvWithGeneralDimensions()
1511 const ConvolutionDimensionNumbers& dimension_numbers, in ConvGeneral()
1529 const ConvolutionDimensionNumbers& dimension_numbers, in ConvGeneralDilated()
1569 const ConvolutionDimensionNumbers& dimension_numbers, in DynamicConvInstruction()
1613 const ConvolutionDimensionNumbers& dimension_numbers, in DynamicConvInputGrad()
1638 const ConvolutionDimensionNumbers& dimension_numbers, in DynamicConvKernelGrad()
1665 const ConvolutionDimensionNumbers& dimension_numbers, in DynamicConvForward()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| H A D | xla.py | 249 dimension_numbers, argument
328 dimension_numbers, argument
571 def gather(operand, start_indices, dimension_numbers, slice_sizes, argument
583 dimension_numbers, indices_are_sorted=False, name=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | image_resize_ops.cc | 252 xla::ConvolutionDimensionNumbers dimension_numbers; in ResizeUsingDilationAndConvolution() local
393 xla::ConvolutionDimensionNumbers dimension_numbers; in ResizeUsingDilationAndConvolutionGradOp() local
604 xla::GatherDimensionNumbers dimension_numbers; in GeneralCompile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| H A D | mlir_hlo_builder.cc | 106 const ConvolutionDimensionNumbers& dimension_numbers, in ConvGeneralDilatedInternal()
383 const GatherDimensionNumbers& dimension_numbers, in GatherInternal()
397 const ScatterDimensionNumbers& dimension_numbers, bool indices_are_sorted, in ScatterInternal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | reference_util_test.cc | 342 ConvolutionDimensionNumbers dimension_numbers; in TEST_F() local
398 ConvolutionDimensionNumbers dimension_numbers; in TEST_F() local
|
| H A D | reference_util.cc | 404 ConvolutionDimensionNumbers dimension_numbers) { in ConvArray4DGeneralDimensions()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | convolution_dimension_numbers_test.cc | 40 ConvolutionDimensionNumbers dimension_numbers; in CreateConvDimensionNumbers() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | cublas_pad_for_gemms.cc | 123 auto dimension_numbers = dot->dot_dimension_numbers(); in CheckCanonical() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | dynamic_dimension_inference.cc | 511 const DotDimensionNumbers& dimension_numbers = dot->dot_dimension_numbers(); in HandleDot() local
596 const ConvolutionDimensionNumbers& dimension_numbers = in HandleConvolution() local
742 const ConvolutionDimensionNumbers& dimension_numbers = in HandleDynamicConvolutionForward() local
|
| H A D | shape_inference.cc | 596 const DotDimensionNumbers& dimension_numbers) { in ValidateDotDimensionNumbers()
647 const DotDimensionNumbers& dimension_numbers, in InferDotOpShape()
|
| H A D | hlo_creation_utils.cc | 120 const ConvolutionDimensionNumbers& dimension_numbers, in MakeConvolveHlo()
|
| H A D | hlo_instructions.cc | 2434 const Window& window, const ConvolutionDimensionNumbers& dimension_numbers, in HloConvolutionInstruction()
3273 const DotDimensionNumbers& dimension_numbers, in HloDotInstruction()
|
| H A D | hlo_instruction.cc | 1173 const Window& window, const ConvolutionDimensionNumbers& dimension_numbers, in CreateConvolve()
1242 const DotDimensionNumbers& dimension_numbers, in CreateDot()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| H A D | xla_ops.cc | 216 xla::DotDimensionNumbers dimension_numbers; in XlaDotShapeFunction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | mlir_hlo_to_hlo.cc | 1478 xla::ScatterDimensionNumbers dimension_numbers = in ExportXlaOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 450 auto dimension_numbers = DotDimensionNumbersAttr::get( in BatchDot() local
3262 auto dimension_numbers = DotDimensionNumbersAttr::get( in matchAndRewrite() local
|