| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 1491 DotDimensionNumbers dot_dnums; in TEST_F() local
1500 DotDimensionNumbers dot_dnums; in TEST_F() local
1513 DotDimensionNumbers dot_dnums; in TEST_F() local
1529 DotDimensionNumbers dot_dnums; in TEST_F() local
1545 DotDimensionNumbers dot_dnums; in TEST_F() local
1561 DotDimensionNumbers dot_dnums; in TEST_F() local
1585 DotDimensionNumbers dot_dnums; in TEST_F() local
1610 DotDimensionNumbers dot_dnums; in TEST_F() local
1632 DotDimensionNumbers dot_dnums; in TEST_F() local
1675 DotDimensionNumbers dot_dnums; in TEST_F() local
[all …]
|
| D | dot_decomposer.cc | 151 DotDimensionNumbers dot_dnums; in CanonicalizeDot() local
|
| D | dot_as_convolution_util.cc | 136 const HloInstruction& conv, const DotConvolutionDimsInfo& dot_dnums, in CreateShardedConvForDotGeneralConvolution()
|
| 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 | 5629 DotDimensionNumbers dot_dnums; in TEST_P() local
5675 DotDimensionNumbers dot_dnums; in TEST_P() local
5721 DotDimensionNumbers dot_dnums; in TEST_P() local
5782 DotDimensionNumbers dot_dnums; in TEST_P() local
5822 DotDimensionNumbers dot_dnums; in TEST_P() local
5876 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 | 4568 DotDimensionNumbers dot_dnums; in TEST_F() local
5234 DotDimensionNumbers dot_dnums; in TEST_P() local
5300 DotDimensionNumbers dot_dnums; in TEST_P() local
5381 DotDimensionNumbers dot_dnums; in TEST_P() local
5446 DotDimensionNumbers dot_dnums; in TEST_P() local
5719 DotDimensionNumbers dot_dnums; in TEST_P() local
5792 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 | 2700 DotDimensionNumbers dot_dnums; in TEST_F() local
|
| D | buffer_assignment_test.cc | 1803 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
|