| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 649 DotDimensionNumbers dnums; in XLA_TYPED_TEST()
684 DotDimensionNumbers dnums; in XLA_TYPED_TEST()
717 DotDimensionNumbers dnums; in XLA_TYPED_TEST()
751 DotDimensionNumbers dnums; in XLA_TYPED_TEST()
958 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
986 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
1016 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
1044 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
1077 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
1110 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 | batch_dot_simplification.cc | 44 const DotDimensionNumbers& dim_numbers = batch_dot->dot_dimension_numbers(); in ElideDegenerateBatchDimensionFromBatchDot()
72 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 | dot_decomposer.cc | 140 DotDimensionNumbers dot_dnums; in CanonicalizeDot()
168 const DotDimensionNumbers& dnums = instruction->dot_dimension_numbers(); in Run()
|
| 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 | 1223 DotDimensionNumbers dot_dnums; in TEST_F()
1232 DotDimensionNumbers dot_dnums; in TEST_F()
1244 DotDimensionNumbers dot_dnums; in TEST_F()
1258 DotDimensionNumbers dot_dnums; in TEST_F()
1272 DotDimensionNumbers dot_dnums; in TEST_F()
1286 DotDimensionNumbers dot_dnums; in TEST_F()
1308 DotDimensionNumbers dot_dnums; in TEST_F()
1332 DotDimensionNumbers dot_dnums; in TEST_F()
1353 DotDimensionNumbers dot_dnums; in TEST_F()
1373 DotDimensionNumbers dot_dnums; in TEST_F()
[all …]
|
| D | hlo_instruction_test.cc | 1152 DotDimensionNumbers dot_dnums; in TEST_F()
1251 DotDimensionNumbers dot_dnums; in TEST_F()
1302 DotDimensionNumbers dot_dnums; in TEST_F()
1383 DotDimensionNumbers dot_dnums; in TEST_F()
1562 DotDimensionNumbers dot_dnums; in TEST_F()
1604 DotDimensionNumbers dot_dnums; in TEST_F()
1660 DotDimensionNumbers dot_dnums; in TEST_F()
1970 DotDimensionNumbers dot_dnums; in TEST_F()
|
| D | hlo_creation_utils.h | 127 const DotDimensionNumbers& dim_numbers,
|
| D | transpose_folding.cc | 83 DotDimensionNumbers new_dim_numbers = dot->dot_dimension_numbers(); in FoldTransposeIntoDot()
|
| 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 | hlo_matchers.cc | 226 const DotDimensionNumbers& dim_nums = instruction->dot_dimension_numbers(); in MatchAndExplain()
|
| D | hlo_cost_analysis_test.cc | 171 DotDimensionNumbers dnums; in TEST_F()
207 DotDimensionNumbers dnums; in TEST_F()
241 DotDimensionNumbers dnums; in TEST_F()
|
| D | heap_simulator_test.cc | 676 DotDimensionNumbers dot_dnums; in TEST_F()
712 DotDimensionNumbers dot_dnums; in TEST_F()
753 DotDimensionNumbers dot_dnums; in TEST_F()
794 DotDimensionNumbers dot_dnums; in TEST_F()
|
| D | hlo_evaluator.h | 127 StatusOr<Literal> EvaluateDotOp(const DotDimensionNumbers& dim_numbers,
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 149 dnums = xla_data_pb2.DotDimensionNumbers()
346 dim_nums = xla_data_pb2.DotDimensionNumbers()
360 dim_nums = xla_data_pb2.DotDimensionNumbers()
373 dim_nums = xla_data_pb2.DotDimensionNumbers()
387 dim_nums = xla_data_pb2.DotDimensionNumbers()
402 dim_nums = xla_data_pb2.DotDimensionNumbers()
|
| /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;
431 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in EmitNaiveLlvmIrGemm()
718 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in GetMatMultDims()
948 Status ValidateDotDimensionNumbers(const DotDimensionNumbers& dim_numbers) { in ValidateDotDimensionNumbers()
1012 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/mlir/xla/ |
| D | hlo_function_importer.h | 114 const DotDimensionNumbers& dnums);
|
| D | hlo_function_importer.cc | 99 DotDimensionNumbers default_dimension_numbers; in DotIsDefault()
744 const DotDimensionNumbers& dnums) { in ConvertDotDimensionNumbers()
762 auto attr = mlir::xla_hlo::DotDimensionNumbers::get( in ConvertDotDimensionNumbers()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | backend_configs.proto | 59 xla.DotDimensionNumbers dot_dimension_numbers = 7;
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | types.h | 302 struct type_caster<xla::DotDimensionNumbers> {
304 PYBIND11_TYPE_CASTER(xla::DotDimensionNumbers, _("xla::DotDimensionNumbers"));
|