Searched refs:mat_mul (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_grad.py | 1157 grad_a = gen_math_ops.mat_mul(grad, b, transpose_b=True)
1158 grad_b = gen_math_ops.mat_mul(a, grad, transpose_a=True)
1160 grad_a = gen_math_ops.mat_mul(grad, b)
1161 grad_b = gen_math_ops.mat_mul(grad, a, transpose_a=True)
1163 grad_a = gen_math_ops.mat_mul(b, grad, transpose_b=True)
1164 grad_b = gen_math_ops.mat_mul(a, grad)
1166 grad_a = gen_math_ops.mat_mul(b, grad, transpose_a=True, transpose_b=True)
1167 grad_b = gen_math_ops.mat_mul(grad, a, transpose_a=True, transpose_b=True)
|
| D | math_ops.py | 2579 return gen_math_ops.mat_mul(
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.h | 85 int64 CountMatMulOperations(const OpInfo& op_info, MatMulDimensions* mat_mul,
|
| D | op_level_cost_estimator.cc | 687 const OpInfo& op_info, MatMulDimensions* mat_mul, in CountMatMulOperations() argument
749 if (mat_mul != nullptr) { in CountMatMulOperations()
750 mat_mul->m = m_dim; in CountMatMulOperations()
751 mat_mul->n = n_dim; in CountMatMulOperations()
752 mat_mul->k = k_dim; in CountMatMulOperations()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | kernelized.py | 205 outputs = gen_math_ops.mat_mul(inputs, self.kernel)
|
| D | core.py | 1029 outputs = gen_math_ops.mat_mul(inputs, self.kernel)
|
| /external/swiftshader/third_party/llvm-7.0/llvm/test/Analysis/Delinearization/ |
| D | constant_functions_multi_dim.ll | 20 define void @mat_mul(float* %C, float* %A, float* %B, i64 %N) #0 !kernel_arg_addr_space !2 !kernel_…
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | benchmarks_test.py | 338 gen_math_ops.mat_mul(m, m, transpose_b=transpose_b)
|