| /external/tensorflow/tensorflow/python/ops/ |
| D | matmul_benchmark.py | 35 def build_graph(device, n, m, k, transpose_a, transpose_b, dtype): argument
51 if not transpose_a:
64 z = math_ops.matmul(x, y, transpose_a=transpose_a, transpose_b=transpose_b)
71 def run_graph(self, device, n, m, k, transpose_a, transpose_b, num_iters, argument
90 output = build_graph(device, n, m, k, transpose_a, transpose_b, dtype)
103 ',ta:' + str(transpose_a) + '.tb:' + str(transpose_b), num_iters,
113 str(transpose_a) + ',tb:' + str(transpose_b)).replace(' ', ''),
118 def run_test_gpu(self, n, m, k, transpose_a, transpose_b, dtype, num_iters): argument
119 self.run_graph(test.gpu_device_name(), n, m, k, transpose_a, transpose_b,
125 for n, m, (transpose_a, transpose_b) in itertools.product(
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| D | csr_sparse_matrix_test.py | 151 def _testSparseSparse(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument
155 dense_shape_a = [5, 13, 7] if transpose_a or adjoint_a else [5, 7, 13]
168 transpose_a=transpose_a,
175 transpose_a=transpose_a,
191 def _testSparseDense(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument
196 dense_shape_a = [5, 13, 7] if transpose_a or adjoint_a else [5, 7, 13]
208 transpose_a=transpose_a,
215 transpose_a=transpose_a,
229 def _testDenseSparse(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument
234 dense_shape_a = [5, 13, 7] if transpose_a or adjoint_a else [5, 7, 13]
[all …]
|
| D | csr_sparse_matrix_dense_mat_mul_grad_test.py | 61 def _testLargeBatchSparseMatrixMatMulGrad(self, datatype, transpose_a, argument
71 if transpose_a or adjoint_a:
88 transpose_a=transpose_a,
|
| D | csr_sparse_matrix_sparse_mat_mul_grad_test.py | 61 def _testLargeBatchSparseMatrixSparseMatMulGrad(self, datatype, transpose_a, argument
71 if transpose_a or adjoint_a:
90 transpose_a=transpose_a,
|
| D | csr_sparse_matrix_ops_test.py | 545 for (transpose_a, transpose_b) in ((False, False), (False, True),
549 if (transpose_a and adjoint_a) or (transpose_b and adjoint_b):
556 if transpose_a or adjoint_a:
568 (transpose_a, transpose_b, adjoint_a, adjoint_b))
575 transpose_a=transpose_a,
582 transpose_a=transpose_a,
607 for (transpose_a, transpose_b) in ((False, False), (False, True),
611 if (transpose_a and adjoint_a) or (transpose_b and adjoint_b):
618 if transpose_a or adjoint_a:
630 (transpose_a, transpose_b, adjoint_a, adjoint_b))
[all …]
|
| /external/tensorflow/tensorflow/python/ops/linalg/sparse/ |
| D | sparse_csr_matrix_grad.py | 117 grad_a_dense = matmul(b, grad, transpose_a=t_b, transpose_b=True)
118 grad_b = sparse_matmul(a, grad, transpose_a=not t_a, transpose_output=t_b)
131 grad_a_dense = matmul(b, grad, transpose_a=True, adjoint_b=True)
134 grad_a_dense = matmul(b, grad, transpose_a=True, transpose_b=True)
143 grad_a_dense = matmul(grad, b, transpose_a=True, transpose_b=not t_b)
145 grad_a_dense = matmul(b, grad, transpose_a=t_b)
147 a, grad, transpose_a=not t_a, transpose_b=True, transpose_output=t_b)
150 grad_a_dense = matmul(grad, b, transpose_a=True, adjoint_b=not adj_b)
161 grad_a_dense = matmul(b, conj(grad), transpose_a=True)
164 grad_a_dense = matmul(b, grad, transpose_a=True)
[all …]
|
| D | sparse_csr_matrix_ops.py | 121 def matmul_shape_inference(a, b, c, transpose_a, transpose_b, adjoint_a, argument
130 transpose_a = transpose_a or adjoint_a
138 c_rows = a_shape.dim[rank - (1 if transpose_a else 2)].size
149 transpose_a=False, argument
185 transpose_a=transpose_a,
202 transpose_a=transpose_a,
211 c_handle = matmul_shape_inference(a_matrix, b_matrix, c, transpose_a,
219 transpose_a=transpose_a,
229 transpose_a=not transpose_b,
230 transpose_b=not transpose_a,
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.cc | 136 bool transpose_a, bool conjugate_a, in SolveWithInvertedDiagonalBlocks() argument
167 bool backward = left_side ^ lower ^ transpose_a; in SolveWithInvertedDiagonalBlocks()
205 if (!left_side ^ transpose_a) { in SolveWithInvertedDiagonalBlocks()
212 remainder = b_row - BatchDot(a_row, transpose_a, x, false, precision); in SolveWithInvertedDiagonalBlocks()
214 remainder = b_row - BatchDot(x, false, a_row, transpose_a, precision); in SolveWithInvertedDiagonalBlocks()
221 BatchDot(inv_block, transpose_a, remainder, false, precision); in SolveWithInvertedDiagonalBlocks()
224 BatchDot(remainder, false, inv_block, transpose_a, precision); in SolveWithInvertedDiagonalBlocks()
373 XlaOp a, XlaOp b, bool left_side, bool lower, bool transpose_a, in SolveByInvertingDiagonalBlocks() argument
404 lower, transpose_a, conjugate_a, in SolveByInvertingDiagonalBlocks()
417 XlaOp a, XlaOp b, bool left_side, bool lower, bool transpose_a, in SolveDirectly() argument
[all …]
|
| D | triangular_solve_expander.h | 42 bool lower, bool transpose_a,
53 bool transpose_a, bool conjugate_a, bool unit_diagonal,
57 bool transpose_a, bool conjugate_a,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | contraction_fusion.mlir | 7 // CHECK-SAME: transpose_a = false, transpose_b = false
8 …%3 = "tf.MatMul"(%arg1, %arg2) {transpose_a = false, transpose_b = false} : (tensor<8x32xf32>, ten…
18 // CHECK-SAME: transpose_a = false, transpose_b = false
19 …%3 = "tf.MatMul"(%arg1, %arg2) {transpose_a = false, transpose_b = false} : (tensor<8x32xf32>, ten…
31 // CHECK-SAME: transpose_a = false, transpose_b = false
32 …%3 = "tf.MatMul"(%arg1, %arg2) {transpose_a = false, transpose_b = false} : (tensor<8x32xf32>, ten…
|
| D | unroll-batch-matmul.mlir | 50 …// CHECK: %[[v26:.*]] = "tf.MatMul"(%[[v2]], %[[v15]]) {transpose_a = false, transpose_b = false} …
51 …// CHECK: %[[v27:.*]] = "tf.MatMul"(%[[v4]], %[[v17]]) {transpose_a = false, transpose_b = false} …
52 …// CHECK: %[[v28:.*]] = "tf.MatMul"(%[[v6]], %[[v19]]) {transpose_a = false, transpose_b = false} …
53 …// CHECK: %[[v29:.*]] = "tf.MatMul"(%[[v8]], %[[v21]]) {transpose_a = false, transpose_b = false} …
54 …// CHECK: %[[v30:.*]] = "tf.MatMul"(%[[v10]], %[[v23]]) {transpose_a = false, transpose_b = false}…
55 …// CHECK: %[[v31:.*]] = "tf.MatMul"(%[[v12]], %[[v25]]) {transpose_a = false, transpose_b = false}…
92 …// CHECK: %[[mm0:.*]] = "tf.MatMul"(%[[v1]], %[[v7]]) {transpose_a = false, transpose_b = false} :…
93 …// CHECK: %[[mm1:.*]] = "tf.MatMul"(%[[v3]], %[[v9]]) {transpose_a = false, transpose_b = false} :…
94 …// CHECK: %[[mm2:.*]] = "tf.MatMul"(%[[v5]], %[[v11]]) {transpose_a = false, transpose_b = false} …
108 …// CHECK: %[[v0:.*]] = "tf.MatMul"(%arg0, %arg1) {transpose_a = false, transpose_b = false} : (ten…
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_matmul_op_test.cc | 90 const bool transpose_a = true; in TEST_F() local
104 .Attr("transpose_a", transpose_a) in TEST_F()
140 const bool transpose_a = true; in TEST_F() local
154 .Attr("transpose_a", transpose_a) in TEST_F()
180 const bool transpose_a = true; in TEST_F() local
194 .Attr("transpose_a", transpose_a) in TEST_F()
273 const bool transpose_a = true; in TEST_F() local
283 .Attr("transpose_a", transpose_a) in TEST_F()
|
| D | matmul_op_test.cc | 101 const Tensor& bias_data, bool transpose_a, in RunMatMulWithBias() argument
110 ops::MatMul::Attrs().TransposeA(transpose_a).TransposeB(transpose_b)); in RunMatMulWithBias()
121 bool transpose_a, bool transpose_b, const string& activation_type, in RunMatMulWithBiasAndActivation() argument
129 ops::MatMul::Attrs().TransposeA(transpose_a).TransposeB(transpose_b)); in RunMatMulWithBiasAndActivation()
152 const std::vector<string>& fused_ops, bool transpose_a, in RunFusedMatMulOp() argument
180 .Attr("transpose_a", transpose_a) in RunFusedMatMulOp()
221 void VerifyMatMulWithBias(int m, int k, int n, bool transpose_a, in VerifyMatMulWithBias() argument
226 RunMatMulWithBias(input_data, filter_data, bias_data, transpose_a, in VerifyMatMulWithBias()
234 transpose_a, transpose_b, out); in VerifyMatMulWithBias()
242 void VerifyConv2DWithBiasAndActivation(int m, int k, int n, bool transpose_a, in VerifyConv2DWithBiasAndActivation() argument
[all …]
|
| D | sparse_matmul_op_test.cc | 51 Node* SparseMatMulNode(Graph* g, Node* in0, Node* in1, bool transpose_a, in SparseMatMulNode() argument
57 .Attr("transpose_a", transpose_a) in SparseMatMulNode()
68 bool transpose_a, bool transpose_b) { in SparseMatMulHelper() argument
72 auto left_shape = transpose_a ? TensorShape({d, m}) : TensorShape({m, d}); in SparseMatMulHelper()
83 test::graph::Constant(g, right), transpose_a, transpose_b, in SparseMatMulHelper()
90 float sparsity_b, bool transpose_a, in SparseMatMul() argument
94 transpose_a, transpose_b); in SparseMatMul()
|
| /external/gemmlowp/eight_bit_int_gemm/ |
| D | eight_bit_int_gemm.cc | 68 template <bool transpose_a, bool transpose_b, bool transpose_c>
84 transpose_a ? MapOrder::RowMajor : MapOrder::ColMajor; in EightBitIntGemmImpl()
107 template <bool transpose_a, bool transpose_b, bool transpose_c>
119 transpose_a ? MapOrder::RowMajor : MapOrder::ColMajor; in EightBitIntGemmInt32Impl()
217 bool CanHandleMetaFastpath(bool transpose_a, bool transpose_b, bool transpose_c, in CanHandleMetaFastpath() argument
226 if (!IsRowMajorOrVector(transpose_a, lda, m, k)) { in CanHandleMetaFastpath()
301 void EightBitIntGemm(bool transpose_a, bool transpose_b, bool transpose_c, in EightBitIntGemm() argument
311 if (CanHandleMetaFastpath(transpose_a, transpose_b, transpose_c, m, n, k, lda, in EightBitIntGemm()
320 if (transpose_a == ta && transpose_b == tb && transpose_c == tc) { \ in EightBitIntGemm()
338 void EightBitIntGemm(bool transpose_a, bool transpose_b, bool transpose_c, in EightBitIntGemm() argument
[all …]
|
| D | eight_bit_int_gemm.h | 55 void EightBitIntGemm(bool transpose_a, bool transpose_b, bool transpose_c,
62 void EightBitIntGemm(bool transpose_a, bool transpose_b, bool transpose_c,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_csr_matrix_ops.cc | 245 bool transpose_a = false; in __anon59a9d6c30702() local
250 TF_RETURN_IF_ERROR(c->GetAttr("transpose_a", &transpose_a)); in __anon59a9d6c30702()
258 if (adjoint_a && transpose_a) { in __anon59a9d6c30702()
266 transpose_a = transpose_a || adjoint_a; in __anon59a9d6c30702()
269 auto output_rows = c->Dim(a_shape, transpose_a ? -1 : -2); in __anon59a9d6c30702()
286 TF_RETURN_IF_ERROR(c->Merge(c->Dim(a_shape, transpose_a ? -2 : -1), in __anon59a9d6c30702()
408 bool transpose_a = false; in __anon59a9d6c30a02() local
410 TF_RETURN_IF_ERROR(c->GetAttr("transpose_a", &transpose_a)); in __anon59a9d6c30a02()
416 if (adjoint_a && transpose_a) { in __anon59a9d6c30a02()
423 transpose_a = transpose_a || adjoint_a; in __anon59a9d6c30a02()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MatMul.pbtxt | 16 name: "transpose_a"
59 name: "transpose_a"
103 name: "transpose_a"
147 name: "transpose_a"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | MatMul.pbtxt | 16 name: "transpose_a"
59 name: "transpose_a"
103 name: "transpose_a"
147 name: "transpose_a"
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | matmul_op_test.py | 165 math_ops.matmul(a, b, transpose_a=True)
257 for adjoint_a, transpose_a in trans_options:
261 transpose_a, adjoint_b, transpose_b)
268 transpose_a=transpose_a,
277 transpose_a=transpose_a,
|
| /external/tensorflow/tensorflow/python/util/ |
| D | function_parameter_canonicalizer_test.py | 49 2, 3, transpose_a=True, name='my_matmul'),
66 transpose_a=True, transpose_b=True, adjoint_a=True)
85 self._matmul_func.canonicalize(2, 3, False, transpose_a=True)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MatMul.pbtxt | 4 name: "transpose_a"
18 "a" (after being transposed if transpose_a is true) must match the
|
| D | api_def_SparseMatrixSparseMatMul.pbtxt | 17 name: "transpose_a"
39 according to the Boolean parameters `transpose_a`, `adjoint_a`, `transpose_b`
40 and `adjoint_b`. At most one of `transpose_a` or `adjoint_a` may be True.
106 transpose_a: If True, `a` transposed before multiplication.
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_op.cc | 92 bool transpose_a = dim_pair[0].first == 0; in Compute() local
99 MklBlasGemm(ctx, transpose_a, transpose_b, m, n, k, a_ptr, in Compute()
100 transpose_a ? m : k, b_ptr, transpose_b ? k : n, c_ptr, n); in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | triangular_solve_test.cc | 448 TriangularSolveOptions::Transpose transpose_a; member
482 /*unit_diagonal=*/false, spec.transpose_a); in XLA_TEST_P()
485 a_tri, spec.transpose_a != TriangularSolveOptions::NO_TRANSPOSE); in XLA_TEST_P()
504 for (TriangularSolveOptions::Transpose transpose_a : in TriangularSolveTests()
510 specs.push_back({dims, left_side, lower, transpose_a}); in TriangularSolveTests()
529 TriangularSolveOptions_Transpose_Name(spec.transpose_a))); in __anon004146d00402()
|