Home
last modified time | relevance | path

Searched defs:transpose_lhs (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_matmul.cc35 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs) { in MatMul()
70 tensorflow::int32 transpose_lhs, in SingleThreadedMatMulDispatch()
90 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF16()
100 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulF32()
109 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulF64()
118 tensorflow::int64 n, tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulC64()
128 tensorflow::int64 n, tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulC128()
138 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulS32()
Druntime_matmul.cc39 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs) { in MatMul()
77 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in MatMulDispatch()
97 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_EigenMatMulF16()
106 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_EigenMatMulF32()
114 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_EigenMatMulF64()
122 tensorflow::int64 n, tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_EigenMatMulC64()
131 tensorflow::int64 n, tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_EigenMatMulC128()
140 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_EigenMatMulS32()
Druntime_matmul_mkl.h49 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_MKLMatMulF32()
58 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_MKLMatMulF64()
67 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_MKLSingleThreadedMatMulF32()
76 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_MKLSingleThreadedMatMulF64()
Druntime_matmul_mkl.cc38 int64 m, int64 n, int64 k, int32 transpose_lhs, in MatMulF32()
60 double* rhs, int64 m, int64 n, int64 k, int32 transpose_lhs, 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()
111 int32 transpose_lhs, in __xla_cpu_runtime_MKLSingleThreadedMatMulF32()
124 int64 k, int32 transpose_lhs, in __xla_cpu_runtime_MKLSingleThreadedMatMulF64()
Dcpu_runtime_test.cc78 bool transpose_lhs, in EigenMatrixMultiply()
139 bool transpose_lhs = std::get<1>(info.param); in Name() local
152 bool transpose_lhs = std::get<1>(GetParam()); in TEST_P() local
176 bool transpose_lhs = std::get<1>(info.param); in Name() local
189 bool transpose_lhs, in MKLMatrixMultiply()
221 bool transpose_lhs = std::get<1>(GetParam()); in TEST_P() local
Ddot_op_emitter.cc829 bool transpose_lhs = !mat_mult_dims.lhs_canonical; in EmitCallToRuntime() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc2560 int64 k, int32 transpose_lhs, int32 transpose_rhs)>& impl_fn) { in MatmulArray2DImpl()
Dalgebraic_simplifier_test.cc5244 bool transpose_lhs, transpose_rhs; in TEST_P() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc539 Value BatchDot(Location loc, Value lhs, bool transpose_lhs, Value rhs, in BatchDot()