| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 659 DotDimensionNumbers dnums; in XLA_TYPED_TEST()
694 DotDimensionNumbers dnums; in XLA_TYPED_TEST()
727 DotDimensionNumbers dnums; in XLA_TYPED_TEST()
761 DotDimensionNumbers dnums; in XLA_TYPED_TEST()
968 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
996 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
1026 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
1054 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
1087 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
1120 DotDimensionNumbers dot_dnums; in XLA_TEST_F()
[all …]
|
| D | multioutput_fusion_test.cc | 87 DotDimensionNumbers dot_dnums; in RunTest2D()
149 DotDimensionNumbers dot_dnums; in RunTest1D()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batch_dot_simplification.cc | 43 const DotDimensionNumbers& dim_numbers = batch_dot->dot_dimension_numbers(); in ElideDegenerateBatchDimensionFromBatchDot()
71 DotDimensionNumbers new_dim_numbers = dim_numbers; in ElideDegenerateBatchDimensionFromBatchDot()
|
| D | dot_merger.cc | 55 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers(); in IsCanonicalDot()
108 const DotDimensionNumbers& dnums_a = a->dot_dimension_numbers(); in TryMergeSameOperand()
109 const DotDimensionNumbers& dnums_b = b->dot_dimension_numbers(); in TryMergeSameOperand()
149 const DotDimensionNumbers& dnums = a->dot_dimension_numbers(); in TryMergeSameOperand()
|
| D | indexed_array_analysis.h | 270 const Shape& shape, const DotDimensionNumbers& dim_numbers,
275 const Shape& shape, const DotDimensionNumbers& dim_numbers,
280 const DotDimensionNumbers& dim_numbers,
|
| D | transpose_folding.cc | 85 DotDimensionNumbers new_dot_dims = dot->dot_dimension_numbers(); in FoldTransposeIntoDot()
261 const DotDimensionNumbers& dot_dims = dot.dot_dimension_numbers(); in IsRowColumnTransposeDotOperand()
|
| D | dot_decomposer.cc | 160 DotDimensionNumbers dot_dnums; in CanonicalizeDot()
197 const DotDimensionNumbers& dnums = instruction->dot_dimension_numbers(); in Run()
|
| D | shape_inference_test.cc | 1505 DotDimensionNumbers dot_dnums; in TEST_F()
1514 DotDimensionNumbers dot_dnums; in TEST_F()
1527 DotDimensionNumbers dot_dnums; in TEST_F()
1543 DotDimensionNumbers dot_dnums; in TEST_F()
1559 DotDimensionNumbers dot_dnums; in TEST_F()
1575 DotDimensionNumbers dot_dnums; in TEST_F()
1599 DotDimensionNumbers dot_dnums; in TEST_F()
1624 DotDimensionNumbers dot_dnums; in TEST_F()
1646 DotDimensionNumbers dot_dnums; in TEST_F()
1689 DotDimensionNumbers dot_dnums; in TEST_F()
[all …]
|
| D | indexed_array_analysis.cc | 1037 const Shape& shape, const DotDimensionNumbers& dim_numbers, in ComputeArrayForDotWithIndexedLhs()
1050 DotDimensionNumbers new_dim_numbers = dim_numbers; in ComputeArrayForDotWithIndexedLhs()
1072 const Shape& shape, const DotDimensionNumbers& dim_numbers, in ComputeArrayForDotWithIndexedRhs()
1086 DotDimensionNumbers new_dim_numbers = dim_numbers; in ComputeArrayForDotWithIndexedRhs()
1107 const Shape& shape, const DotDimensionNumbers& dim_numbers, in ComputeArrayForDot()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 211 dnums = xla_data_pb2.DotDimensionNumbers()
249 dnums = xla_data_pb2.DotDimensionNumbers()
691 dim_nums = xla_data_pb2.DotDimensionNumbers()
704 dim_nums = xla_data_pb2.DotDimensionNumbers()
718 dim_nums = xla_data_pb2.DotDimensionNumbers()
730 dim_nums = xla_data_pb2.DotDimensionNumbers()
744 dim_nums = xla_data_pb2.DotDimensionNumbers()
758 dim_nums = xla_data_pb2.DotDimensionNumbers()
769 dim_nums = xla_data_pb2.DotDimensionNumbers()
780 dim_nums = xla_data_pb2.DotDimensionNumbers()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.pyi | 124 class DotDimensionNumbers:
132 dimension_numbers: Union[DotDimensionNumbers,
135 ) -> DotDimensionNumbers:
|
| D | xla_client.py | 465 class DotDimensionNumbers: class
478 dimension_numbers: Union[DotDimensionNumbers, argument
481 ) -> DotDimensionNumbers:
495 dot_dims_proto = DotDimensionNumbers()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | jitrt_custom_calls.h | 38 struct DotDimensionNumbers { struct
84 xla::gpu::DotDimensionNumbers,
|
| D | gemm_broadcast_folding_rewriter.cc | 51 DotDimensionNumbers *dim_nums = config.mutable_dot_dimension_numbers(); in HandleCustomCall()
|
| D | gemm_rewriter.cc | 55 const DotDimensionNumbers &dot_dims = config.dot_dimension_numbers(); in SetName()
312 const DotDimensionNumbers &dot_dims = config.dot_dimension_numbers(); in FuseVectorBiasAdd()
|
| D | backend_configs.proto | 57 xla.DotDimensionNumbers dot_dimension_numbers = 7;
|
| D | gpu_layout_assignment.cc | 232 const DotDimensionNumbers& dot_dims = dot->dot_dimension_numbers(); in DotCanSupportShapeWithLayout()
265 const DotDimensionNumbers& dot_dims = in AddBackendConstraints()
|
| /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;
618 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in EmitNaiveLlvmIrGemm()
1008 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in GetMatMultDims()
1038 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in GetBatchMatMultDims()
1289 Status ValidateDotDimensionNumbers(const DotDimensionNumbers& dim_numbers) { in ValidateDotDimensionNumbers()
1346 DotDimensionNumbers adjusted_dim_numbers = dot.dot_dimension_numbers(); in PotentiallyImplementedAsEigenMatmul()
1425 DotDimensionNumbers adjusted_dim_numbers = in EmitBatchDotOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_importer.h | 44 const DotDimensionNumbers& dnums, mlir::Builder* builder);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_dot_op.cc | 64 xla::DotDimensionNumbers dnums_;
|
| D | xla_broadcast_helper_op.cc | 102 xla::DotDimensionNumbers dnums_;
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/lhlo_gpu/IR/ |
| D | lhlo_gpu_ops.td | 134 DotDimensionNumbers:$dot_dimension_numbers,
149 DotDimensionNumbers:$dot_dimension_numbers,
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops_base_attrs.td | 48 def DotDimensionNumbers : AttrDef<HLO_Dialect, "DotDimensionNumbers"> {
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/python/ |
| D | attributes.py | 74 attr = DotDimensionNumbers.get(
|