| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_dimension_numbers_test.cc | 43 ConvolutionDimensionNumbers dimension_numbers; in CreateConvDimensionNumbers() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_resize_ops.cc | 241 xla::ConvolutionDimensionNumbers dimension_numbers; in ResizeUsingDilationAndConvolution() local
364 xla::ConvolutionDimensionNumbers dimension_numbers; in ResizeUsingDilationAndConvolutionGradOp() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 582 const DotDimensionNumbers& dimension_numbers) { in DotGeneral()
591 const ConvolutionDimensionNumbers& dimension_numbers, in ConvGeneralDilated()
756 const GatherDimensionNumbers& dimension_numbers, in Gather()
765 const ScatterDimensionNumbers& dimension_numbers) { in Scatter()
|
| D | xla_client.py | 1604 def DotGeneral(self, lhs, rhs, dimension_numbers): argument
1675 rhs_dilation, dimension_numbers=None, argument
1775 def Gather(self, a, start_indices, dimension_numbers, slice_sizes): argument
1780 dimension_numbers): argument
1919 def GetDotDimensionsFromLists(dimension_numbers): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1015 DotDimensionNumbers dimension_numbers; in Dot() local
1024 const DotDimensionNumbers& dimension_numbers, in DotGeneral()
1121 Padding padding, const ConvolutionDimensionNumbers& dimension_numbers, in ConvWithGeneralDimensions()
1158 const ConvolutionDimensionNumbers& dimension_numbers, in ConvGeneral()
1170 const ConvolutionDimensionNumbers& dimension_numbers, in ConvGeneralDilated()
1841 const GatherDimensionNumbers& dimension_numbers, in Gather()
1867 const ScatterDimensionNumbers& dimension_numbers) { in Scatter()
2647 ConvolutionDimensionNumbers dimension_numbers; in CreateDefaultConvDimensionNumbers() local
2964 const DotDimensionNumbers& dimension_numbers, in DotGeneral()
2991 Padding padding, const ConvolutionDimensionNumbers& dimension_numbers, in ConvWithGeneralDimensions()
[all …]
|
| /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 | 468 ConvolutionDimensionNumbers dimension_numbers) { in ConvArray4DGeneralDimensions()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 234 dimension_numbers, argument
282 def dot_general(lhs, rhs, dimension_numbers, precision_config=None, name=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference.cc | 205 const DotDimensionNumbers& dimension_numbers = in HandleDot() local
267 const ConvolutionDimensionNumbers& dimension_numbers = in HandleConvolution() local
|
| D | hlo_creation_utils.cc | 84 const Window& window, const ConvolutionDimensionNumbers& dimension_numbers, in MakeConvolveHlo()
|
| D | shape_inference.cc | 559 const DotDimensionNumbers& dimension_numbers) { in ValidateDotDimensionNumbers()
610 const DotDimensionNumbers& dimension_numbers) { in InferDotOpShape()
|
| D | hlo_instruction.cc | 783 const ConvolutionDimensionNumbers& dimension_numbers, in CreateConvolve()
817 const DotDimensionNumbers& dimension_numbers, in CreateDot()
|
| D | hlo_instructions.cc | 1857 const ConvolutionDimensionNumbers& dimension_numbers, in HloConvolutionInstruction()
2511 const DotDimensionNumbers& dimension_numbers, in HloDotInstruction()
|