| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 1470 DotDimensionNumbers dot_dnums; in TEST_F() local
1479 DotDimensionNumbers dot_dnums; in TEST_F() local
1492 DotDimensionNumbers dot_dnums; in TEST_F() local
1508 DotDimensionNumbers dot_dnums; in TEST_F() local
1524 DotDimensionNumbers dot_dnums; in TEST_F() local
1540 DotDimensionNumbers dot_dnums; in TEST_F() local
1564 DotDimensionNumbers dot_dnums; in TEST_F() local
1589 DotDimensionNumbers dot_dnums; in TEST_F() local
1611 DotDimensionNumbers dot_dnums; in TEST_F() local
1654 DotDimensionNumbers dot_dnums; in TEST_F() local
[all …]
|
| D | dot_as_convolution_util.cc | 110 const HloInstruction& conv, const DotConvolutionDimsInfo& dot_dnums, in CreateShardedConvForDotGeneralConvolution()
|
| D | dot_decomposer.cc | 151 DotDimensionNumbers dot_dnums; in CanonicalizeDot() local
|
| D | hlo_instruction_test.cc | 1210 DotDimensionNumbers dot_dnums; in TEST_F() local
1309 DotDimensionNumbers dot_dnums; in TEST_F() local
1360 DotDimensionNumbers dot_dnums; in TEST_F() local
1441 DotDimensionNumbers dot_dnums; in TEST_F() local
1620 DotDimensionNumbers dot_dnums; in TEST_F() local
1662 DotDimensionNumbers dot_dnums; in TEST_F() local
1718 DotDimensionNumbers dot_dnums; in TEST_F() local
2028 DotDimensionNumbers dot_dnums; in TEST_F() local
|
| D | hlo_computation_test.cc | 550 DotDimensionNumbers dot_dnums; in TEST_F() local
585 DotDimensionNumbers dot_dnums; in TEST_F() local
621 DotDimensionNumbers dot_dnums; in TEST_F() local
|
| D | dynamic_dimension_inference_test.cc | 322 DotDimensionNumbers dot_dnums; in TEST_F() local
363 DotDimensionNumbers dot_dnums; in TEST_F() local
402 DotDimensionNumbers dot_dnums; in TEST_F() local
|
| D | heap_simulator_test.cc | 681 DotDimensionNumbers dot_dnums; in TEST_F() local
717 DotDimensionNumbers dot_dnums; in TEST_F() local
758 DotDimensionNumbers dot_dnums; in TEST_F() local
799 DotDimensionNumbers dot_dnums; in TEST_F() local
|
| D | bfloat16_normalization_test.cc | 410 DotDimensionNumbers dot_dnums; in TEST_F() local
|
| D | memory_space_assignment_test.cc | 4838 DotDimensionNumbers dot_dnums; in TEST_P() local
4884 DotDimensionNumbers dot_dnums; in TEST_P() local
4930 DotDimensionNumbers dot_dnums; in TEST_P() local
4991 DotDimensionNumbers dot_dnums; in TEST_P() local
5031 DotDimensionNumbers dot_dnums; in TEST_P() local
5085 DotDimensionNumbers dot_dnums; in TEST_P() local
|
| D | hlo_evaluator_test.cc | 812 DotDimensionNumbers dot_dnums; in TEST_P() local
858 DotDimensionNumbers dot_dnums; in TEST_P() local
902 DotDimensionNumbers dot_dnums; in TEST_P() local
939 DotDimensionNumbers dot_dnums; in TEST_P() local
|
| D | algebraic_simplifier_test.cc | 4535 DotDimensionNumbers dot_dnums; in TEST_F() local
5201 DotDimensionNumbers dot_dnums; in TEST_P() local
5267 DotDimensionNumbers dot_dnums; in TEST_P() local
5348 DotDimensionNumbers dot_dnums; in TEST_P() local
5413 DotDimensionNumbers dot_dnums; in TEST_P() local
5630 DotDimensionNumbers dot_dnums; in TEST_P() local
5703 DotDimensionNumbers dot_dnums; in TEST_P() local
|
| D | bfloat16_propagation_test.cc | 88 DotDimensionNumbers dot_dnums; in CreateDot() local
|
| D | hlo_dataflow_analysis_test.cc | 2622 DotDimensionNumbers dot_dnums; in TEST_F() local
|
| D | buffer_assignment_test.cc | 1764 DotDimensionNumbers dot_dnums; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 967 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
995 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
1025 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
1053 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
1086 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
1119 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
1144 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
1169 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
1188 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
|
| D | multioutput_fusion_test.cc | 89 DotDimensionNumbers dot_dnums; in RunTest2D() local
151 DotDimensionNumbers dot_dnums; in RunTest1D() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 145 DotDimensionNumbers dot_dnums; in ShuffleWithinEachPartitionUsingOneHot() local
|
| D | convolution_handler.cc | 895 const dot_as_convolution_util::DotConvolutionDimsInfo& dot_dnums, in CreateShardedConvConvolution()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 40 DotDimensionNumbers dot_dnums; in MakeDot() local
|