| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_gather_to_torch_index_select.cc | 51 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 | 180 auto dimension_numbers = mhlo::DotDimensionNumbers::get( in matchAndRewrite() local
|
| D | legalize_to_linalg.cc | 178 const DimensionNumbersTy& dimension_numbers) { in HasCanonicalDimensionNumbers()
2159 if (const mhlo::ConvDimensionNumbers& dimension_numbers = in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 252 dimension_numbers, argument
322 dimension_numbers, argument
550 def gather(operand, start_indices, dimension_numbers, slice_sizes, argument
562 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/client/ |
| D | xla_builder.cc | 1302 DotDimensionNumbers dimension_numbers; in Dot() local
1312 XlaOp lhs, XlaOp rhs, const DotDimensionNumbers& dimension_numbers, in DotGeneral()
1329 const DotDimensionNumbers& dimension_numbers, in DotGeneralInternal()
1411 Padding padding, const ConvolutionDimensionNumbers& dimension_numbers, in ConvWithGeneralDimensions()
1450 const ConvolutionDimensionNumbers& dimension_numbers, in ConvGeneral()
1464 const ConvolutionDimensionNumbers& dimension_numbers, in ConvGeneralDilated()
1502 const ConvolutionDimensionNumbers& dimension_numbers, in DynamicConvInstruction()
1544 const ConvolutionDimensionNumbers& dimension_numbers, in DynamicConvInputGrad()
1567 const ConvolutionDimensionNumbers& dimension_numbers, in DynamicConvKernelGrad()
1593 const ConvolutionDimensionNumbers& dimension_numbers, in DynamicConvForward()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cublas_pad_for_gemms.cc | 123 auto dimension_numbers = dot->dot_dimension_numbers(); in CheckCanonical() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_resize_ops.cc | 247 xla::ConvolutionDimensionNumbers dimension_numbers; in ResizeUsingDilationAndConvolution() local
388 xla::ConvolutionDimensionNumbers dimension_numbers; in ResizeUsingDilationAndConvolutionGradOp() local
|
| /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 | 403 ConvolutionDimensionNumbers dimension_numbers) { in ConvArray4DGeneralDimensions()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 98 const ConvolutionDimensionNumbers& dimension_numbers, in ConvGeneralDilatedInternal()
306 const GatherDimensionNumbers& dimension_numbers, in GatherInternal()
320 const ScatterDimensionNumbers& dimension_numbers, bool indices_are_sorted, in ScatterInternal()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 213 xla::DotDimensionNumbers dimension_numbers; in XlaDotShapeFunction() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference.cc | 479 const DotDimensionNumbers& dimension_numbers = in HandleDot() local
567 const ConvolutionDimensionNumbers& dimension_numbers = in HandleConvolution() local
713 const ConvolutionDimensionNumbers& dimension_numbers = in HandleDynamicConvolutionForward() local
|
| D | hlo_creation_utils.cc | 96 const ConvolutionDimensionNumbers& dimension_numbers, in MakeConvolveHlo()
|
| D | shape_inference.cc | 588 const DotDimensionNumbers& dimension_numbers) { in ValidateDotDimensionNumbers()
639 const DotDimensionNumbers& dimension_numbers, in InferDotOpShape()
|
| D | hlo_instruction.cc | 1103 const Window& window, const ConvolutionDimensionNumbers& dimension_numbers, in CreateConvolve()
1145 const DotDimensionNumbers& dimension_numbers, in CreateDot()
|
| D | hlo_instructions.cc | 2274 const Window& window, const ConvolutionDimensionNumbers& dimension_numbers, in HloConvolutionInstruction()
3114 const DotDimensionNumbers& dimension_numbers, in HloDotInstruction()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 1066 xla::ScatterDimensionNumbers dimension_numbers = in ExportXlaOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 500 auto dimension_numbers = DotDimensionNumbers::get( in BatchDot() local
3465 auto dimension_numbers = DotDimensionNumbers::get( in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 400 auto dimension_numbers = op->dimension_numbers(); in GatherShapeInferImpl() local
|