| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 614 DotDimensionNumbers dnums; in XLA_TYPED_TEST()
651 DotDimensionNumbers dnums; in XLA_TYPED_TEST()
685 DotDimensionNumbers dnums; in XLA_TYPED_TEST()
720 DotDimensionNumbers dnums; in XLA_TYPED_TEST()
927 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
955 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
985 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
1013 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
1046 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
1079 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
[all …]
|
| D | multioutput_fusion_test.cc | 89 DotDimensionNumbers dot_dnums; in RunTest2D()
151 DotDimensionNumbers dot_dnums; in RunTest1D()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_decomposer.cc | 36 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers(); in DecomposeBatchDot()
136 DotDimensionNumbers dot_dnums; in DecomposeBatchDot()
269 DotDimensionNumbers dot_dnums; in CanonicalizeDot()
297 const DotDimensionNumbers& dnums = instruction->dot_dimension_numbers(); in Run()
334 const DotDimensionNumbers& dnums = instruction->dot_dimension_numbers(); in Run()
|
| D | batch_dot_simplification.cc | 26 const DotDimensionNumbers& dim_numbers = batch_dot->dot_dimension_numbers(); in ElideDegenerateBatchDimensionFromBatchDot()
54 DotDimensionNumbers new_dim_numbers = dim_numbers; in ElideDegenerateBatchDimensionFromBatchDot()
|
| D | indexed_array_analysis.h | 269 const Shape& shape, const DotDimensionNumbers& dim_numbers,
274 const Shape& shape, const DotDimensionNumbers& dim_numbers,
279 const DotDimensionNumbers& dim_numbers,
|
| D | indexed_array_analysis.cc | 1033 const Shape& shape, const DotDimensionNumbers& dim_numbers, in ComputeArrayForDotWithIndexedLhs()
1046 DotDimensionNumbers new_dim_numbers = dim_numbers; in ComputeArrayForDotWithIndexedLhs()
1068 const Shape& shape, const DotDimensionNumbers& dim_numbers, in ComputeArrayForDotWithIndexedRhs()
1082 DotDimensionNumbers new_dim_numbers = dim_numbers; in ComputeArrayForDotWithIndexedRhs()
1103 const Shape& shape, const DotDimensionNumbers& dim_numbers, in ComputeArrayForDot()
|
| D | shape_inference_test.cc | 965 DotDimensionNumbers dot_dnums; in TEST_F()
977 DotDimensionNumbers dot_dnums; in TEST_F()
989 DotDimensionNumbers dot_dnums; in TEST_F()
1003 DotDimensionNumbers dot_dnums; in TEST_F()
1017 DotDimensionNumbers dot_dnums; in TEST_F()
1031 DotDimensionNumbers dot_dnums; in TEST_F()
1053 DotDimensionNumbers dot_dnums; in TEST_F()
1077 DotDimensionNumbers dot_dnums; in TEST_F()
1098 DotDimensionNumbers dot_dnums; in TEST_F()
1118 DotDimensionNumbers dot_dnums; in TEST_F()
[all …]
|
| D | hlo_cost_analysis_test.cc | 167 DotDimensionNumbers dnums; in TEST_F()
196 DotDimensionNumbers dnums; in TEST_F()
223 DotDimensionNumbers dnums; in TEST_F()
|
| D | hlo_creation_utils.h | 109 const DotDimensionNumbers& dim_numbers,
|
| D | hlo_instruction_test.cc | 1149 DotDimensionNumbers dot_dnums; in TEST_F()
1190 DotDimensionNumbers dot_dnums; in TEST_F()
1241 DotDimensionNumbers dot_dnums; in TEST_F()
1322 DotDimensionNumbers dot_dnums; in TEST_F()
1487 DotDimensionNumbers dot_dnums; in TEST_F()
1529 DotDimensionNumbers dot_dnums; in TEST_F()
1585 DotDimensionNumbers dot_dnums; in TEST_F()
|
| D | hlo_matchers.cc | 222 const DotDimensionNumbers& dim_nums = instruction->dot_dimension_numbers(); in MatchAndExplain()
|
| D | hlo_computation_test.cc | 550 DotDimensionNumbers dot_dnums; in TEST_F()
585 DotDimensionNumbers dot_dnums; in TEST_F()
621 DotDimensionNumbers dot_dnums; in TEST_F()
|
| D | transpose_folding.cc | 83 DotDimensionNumbers new_dim_numbers = dot->dot_dimension_numbers(); in FoldTransposeIntoDot()
|
| D | shape_inference.h | 277 const DotDimensionNumbers& dimension_numbers);
|
| D | hlo_evaluator.h | 125 StatusOr<Literal> EvaluateDotOp(const DotDimensionNumbers& dim_numbers,
|
| D | heap_simulator_test.cc | 491 DotDimensionNumbers dot_dnums; in TEST_F()
527 DotDimensionNumbers dot_dnums; in TEST_F()
565 DotDimensionNumbers dot_dnums; in TEST_F()
606 DotDimensionNumbers dot_dnums; in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter_internal.h | 41 DotDimensionNumbers dim_nums;
|
| D | dot_op_emitter.cc | 60 DotDimensionNumbers dim_nums;
411 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in EmitNaiveLlvmIrGemm()
689 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in GetMatMultDims()
896 Status ValidateDotDimensionNumbers(const DotDimensionNumbers& dim_numbers) { in ValidateDotDimensionNumbers()
960 DotDimensionNumbers adjusted_dim_numbers = dot.dot_dimension_numbers(); in EmitBatchDotOperation()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_dot_op.cc | 56 xla::DotDimensionNumbers dnums_;
|
| D | xla_broadcast_helper_op.cc | 102 xla::DotDimensionNumbers dnums_;
|
| D | resampler_ops.cc | 401 xla::DotDimensionNumbers dot_dims; in CalculateGradWarp()
538 xla::DotDimensionNumbers dot_dims; in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.cc | 271 DotDimensionNumbers GetDimensionNumbers(const HloInstruction& hlo_instruction) { in GetDimensionNumbers()
323 DotDimensionNumbers dim_nums = GetDimensionNumbers(*hlo_instruction()); in ExecuteOnStream()
|
| D | ir_emission_utils.cc | 69 const DotDimensionNumbers& dim_numbers = dot.dot_dimension_numbers(); in DotImplementedAsGemm()
|
| D | gpu_layout_assignment.cc | 194 const DotDimensionNumbers& dim_nums = in AddBackendConstraints()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_data.i | 451 %typemap(in) const DotDimensionNumbers&
452 (DotDimensionNumbers dimension_numbers) {
|