| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_importer.h | 41 mlir::mhlo::DotDimensionNumbers ConvertDotDimensionNumbers(
42 const DotDimensionNumbers& dnums, mlir::Builder* builder);
|
| D | attribute_importer.cc | 80 mlir::mhlo::DotDimensionNumbers ConvertDotDimensionNumbers( in ConvertDotDimensionNumbers()
81 const DotDimensionNumbers& dnums, mlir::Builder* builder) { in ConvertDotDimensionNumbers()
99 return mlir::mhlo::DotDimensionNumbers::get( in ConvertDotDimensionNumbers()
|
| /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 | 151 DotDimensionNumbers dot_dnums; in CanonicalizeDot()
180 const DotDimensionNumbers& dnums = instruction->dot_dimension_numbers(); in Run()
|
| D | shape_inference_test.cc | 1470 DotDimensionNumbers dot_dnums; in TEST_F()
1479 DotDimensionNumbers dot_dnums; in TEST_F()
1492 DotDimensionNumbers dot_dnums; in TEST_F()
1508 DotDimensionNumbers dot_dnums; in TEST_F()
1524 DotDimensionNumbers dot_dnums; in TEST_F()
1540 DotDimensionNumbers dot_dnums; in TEST_F()
1564 DotDimensionNumbers dot_dnums; in TEST_F()
1589 DotDimensionNumbers dot_dnums; in TEST_F()
1611 DotDimensionNumbers dot_dnums; in TEST_F()
1654 DotDimensionNumbers dot_dnums; in TEST_F()
[all …]
|
| 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 | hlo_instruction_test.cc | 1210 DotDimensionNumbers dot_dnums; in TEST_F()
1309 DotDimensionNumbers dot_dnums; in TEST_F()
1360 DotDimensionNumbers dot_dnums; in TEST_F()
1441 DotDimensionNumbers dot_dnums; in TEST_F()
1620 DotDimensionNumbers dot_dnums; in TEST_F()
1662 DotDimensionNumbers dot_dnums; in TEST_F()
1718 DotDimensionNumbers dot_dnums; in TEST_F()
2028 DotDimensionNumbers dot_dnums; in TEST_F()
|
| D | transpose_folding.cc | 112 DotDimensionNumbers new_dim_numbers = dot->dot_dimension_numbers(); in FoldTransposeIntoDot()
|
| D | hlo_matchers.cc | 226 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 | hlo_creation_utils.h | 139 const DotDimensionNumbers& dim_numbers,
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 658 DotDimensionNumbers dnums; in XLA_TYPED_TEST()
693 DotDimensionNumbers dnums; in XLA_TYPED_TEST()
726 DotDimensionNumbers dnums; in XLA_TYPED_TEST()
760 DotDimensionNumbers dnums; in XLA_TYPED_TEST()
967 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
995 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
1025 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
1053 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
1086 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
1119 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/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops_base_structs.td | 23 def DotDimensionNumbers : StructAttr<"DotDimensionNumbers", HLO_Dialect, [
|
| D | lhlo_gpu_ops.td | 174 DotDimensionNumbers:$dot_dimension_numbers,
188 DotDimensionNumbers:$dot_dimension_numbers,
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 490 class DotDimensionNumbers(object): class
503 dimension_numbers: Union[DotDimensionNumbers, argument
506 ) -> DotDimensionNumbers:
520 dot_dims_proto = DotDimensionNumbers()
|
| D | types.h | 258 struct type_caster<xla::DotDimensionNumbers> {
260 PYBIND11_TYPE_CASTER(xla::DotDimensionNumbers, _("xla::DotDimensionNumbers"));
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 173 dnums = xla_data_pb2.DotDimensionNumbers()
550 dim_nums = xla_data_pb2.DotDimensionNumbers()
564 dim_nums = xla_data_pb2.DotDimensionNumbers()
577 dim_nums = xla_data_pb2.DotDimensionNumbers()
591 dim_nums = xla_data_pb2.DotDimensionNumbers()
606 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 | 72 DotDimensionNumbers dim_nums;
554 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in EmitNaiveLlvmIrGemm()
855 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in GetMatMultDims()
1092 Status ValidateDotDimensionNumbers(const DotDimensionNumbers& dim_numbers) { in ValidateDotDimensionNumbers()
1156 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_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | backend_configs.proto | 59 xla.DotDimensionNumbers dot_dimension_numbers = 7;
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_affine.cc | 62 mhlo::DotDimensionNumbers dot_dimension_numbers = in matchAndRewrite()
|