| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 1505 DotDimensionNumbers dot_dnums; in TEST_F() local
1514 DotDimensionNumbers dot_dnums; in TEST_F() local
1527 DotDimensionNumbers dot_dnums; in TEST_F() local
1543 DotDimensionNumbers dot_dnums; in TEST_F() local
1559 DotDimensionNumbers dot_dnums; in TEST_F() local
1575 DotDimensionNumbers dot_dnums; in TEST_F() local
1599 DotDimensionNumbers dot_dnums; in TEST_F() local
1624 DotDimensionNumbers dot_dnums; in TEST_F() local
1646 DotDimensionNumbers dot_dnums; in TEST_F() local
1689 DotDimensionNumbers dot_dnums; in TEST_F() local
[all …]
|
| D | dot_decomposer.cc | 160 DotDimensionNumbers dot_dnums; in CanonicalizeDot() local
|
| D | dot_as_convolution_util.cc | 137 const HloInstruction& conv, const DotConvolutionDimsInfo& dot_dnums, in CreateShardedConvForDotGeneralConvolution()
|
| D | hlo_instruction_test.cc | 1309 DotDimensionNumbers dot_dnums; in TEST_F() local
1408 DotDimensionNumbers dot_dnums; in TEST_F() local
1459 DotDimensionNumbers dot_dnums; in TEST_F() local
1540 DotDimensionNumbers dot_dnums; in TEST_F() local
1787 DotDimensionNumbers dot_dnums; in TEST_F() local
1834 DotDimensionNumbers dot_dnums; in TEST_F() local
1890 DotDimensionNumbers dot_dnums; in TEST_F() local
2200 DotDimensionNumbers dot_dnums; in TEST_F() local
|
| D | hlo_computation_test.cc | 532 DotDimensionNumbers dot_dnums; in TEST_F() local
568 DotDimensionNumbers dot_dnums; in TEST_F() local
605 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 | dynamic_dimension_inference_test.cc | 329 DotDimensionNumbers dot_dnums; in TEST_F() local
370 DotDimensionNumbers dot_dnums; in TEST_F() local
409 DotDimensionNumbers dot_dnums; in TEST_F() local
|
| D | memory_space_assignment_test.cc | 6069 DotDimensionNumbers dot_dnums; in TEST_P() local
6111 DotDimensionNumbers dot_dnums; in TEST_P() local
6153 DotDimensionNumbers dot_dnums; in TEST_P() local
6199 DotDimensionNumbers dot_dnums; in TEST_P() local
6245 DotDimensionNumbers dot_dnums; in TEST_P() local
6302 DotDimensionNumbers dot_dnums; in TEST_P() local
6340 DotDimensionNumbers dot_dnums; in TEST_P() local
6377 DotDimensionNumbers dot_dnums; in TEST_P() local
6428 DotDimensionNumbers dot_dnums; in TEST_P() local
6478 DotDimensionNumbers dot_dnums; in TEST_P() local
[all …]
|
| D | bfloat16_normalization_test.cc | 410 DotDimensionNumbers dot_dnums; in TEST_F() local
|
| D | hlo_evaluator_test.cc | 850 DotDimensionNumbers dot_dnums; in TEST_P() local
896 DotDimensionNumbers dot_dnums; in TEST_P() local
940 DotDimensionNumbers dot_dnums; in TEST_P() local
977 DotDimensionNumbers dot_dnums; in TEST_P() local
|
| D | algebraic_simplifier_test.cc | 4948 DotDimensionNumbers dot_dnums; in TEST_F() local
5842 DotDimensionNumbers dot_dnums; in TEST_P() local
5908 DotDimensionNumbers dot_dnums; in TEST_P() local
5989 DotDimensionNumbers dot_dnums; in TEST_P() local
6054 DotDimensionNumbers dot_dnums; in TEST_P() local
6327 DotDimensionNumbers dot_dnums; in TEST_P() local
6400 DotDimensionNumbers dot_dnums; in TEST_P() local
|
| D | bfloat16_propagation_test.cc | 89 DotDimensionNumbers dot_dnums; in CreateDot() local
|
| D | hlo_dataflow_analysis_test.cc | 2683 DotDimensionNumbers dot_dnums; in TEST_F() local
|
| D | buffer_assignment_test.cc | 1800 DotDimensionNumbers dot_dnums; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 968 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
996 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
1026 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
1054 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
1087 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
1120 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
1145 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
1170 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
1189 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
|
| D | multioutput_fusion_test.cc | 87 DotDimensionNumbers dot_dnums; in RunTest2D() local
149 DotDimensionNumbers dot_dnums; in RunTest1D() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 143 DotDimensionNumbers dot_dnums; in ShuffleWithinEachPartitionUsingOneHot() local
|
| D | convolution_handler.cc | 897 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
|