Home
last modified time | relevance | path

Searched defs:transpose_rhs (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_matmul_mkl.h50 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF32()
59 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF64()
68 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_MKLSingleThreadedMatMulF32()
77 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_MKLSingleThreadedMatMulF64()
Druntime_matmul_mkl.cc39 int32 transpose_rhs) { in MatMulF32()
61 int32 transpose_rhs) { in MatMulF64()
80 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF32()
95 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF64()
112 int32 transpose_rhs) { in __xla_cpu_runtime_MKLSingleThreadedMatMulF32()
125 int32 transpose_rhs) { in __xla_cpu_runtime_MKLSingleThreadedMatMulF64()
Druntime_matmul.cc41 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) { in MatMul()
78 int32 transpose_rhs) { in MatMulDispatch()
102 int32 transpose_rhs) { in __xla_cpu_runtime_EigenMatMulF16()
109 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) { in __xla_cpu_runtime_EigenMatMulF32()
116 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) { in __xla_cpu_runtime_EigenMatMulF64()
Druntime_single_threaded_matmul.cc38 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) { in MatMul()
72 int32 transpose_lhs, int32 transpose_rhs) { in SingleThreadedMatMulDispatch()
95 int32 transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulF16()
105 int32 transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulF32()
115 int32 transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulF64()
Dcpu_runtime_test.cc80 bool transpose_rhs, in EigenMatrixMultiply()
142 bool transpose_rhs = std::get<2>(info.param); in Name() local
155 bool transpose_rhs = std::get<2>(GetParam()); in TEST_P() local
179 bool transpose_rhs = std::get<2>(info.param); in Name() local
192 bool transpose_rhs, in MKLMatrixMultiply()
224 bool transpose_rhs = std::get<2>(GetParam()); in TEST_P() local
Druntime_matvec.h65 int32 transpose_lhs, int32 transpose_rhs) { in DispatchMatVec()
113 tensorflow::int32 transpose_rhs) { in EigenMatVec()
Ddot_op_emitter.cc664 bool transpose_rhs = mat_mult_dims.rhs_non_canonical; in EmitCallToRuntime() local
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinalg_impl.py339 transpose_rhs=False, argument
478 transpose_rhs=False, argument
/external/tensorflow/tensorflow/python/kernel_tests/
Dtridiagonal_solve_op_test.py57 transpose_rhs=False, argument
69 transpose_rhs=False, argument
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc4387 bool transpose_lhs, transpose_rhs; in TEST_P() local