Home
last modified time | relevance | path

Searched refs:MatMulOp (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmatmul_op.cc30 class MatMulOp : public XlaOpKernel { class
32 explicit MatMulOp(OpKernelConstruction* ctx, bool is_sparse = false) in MatMulOp() function in tensorflow::__anondc94a7a40111::MatMulOp
50 ~MatMulOp() override = default;
97 REGISTER_XLA_OP(Name("MatMul").TypeConstraint("T", kMatmulTypes), MatMulOp);
99 class SparseMatMulOp : public MatMulOp {
101 explicit SparseMatMulOp(OpKernelConstruction* ctx) : MatMulOp(ctx, true) {} in SparseMatMulOp()
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op.cc439 class MatMulOp : public OpKernel { class
441 explicit MatMulOp(OpKernelConstruction* ctx) in MatMulOp() function in tensorflow::MatMulOp
563 MatMulOp<CPUDevice, T, false /* cublas, ignored for CPU */>);
568 MatMulOp<CPUDevice, T, false /* cublas, ignored for CPU */>); \
574 MatMulOp<GPUDevice, T, true /* cublas, true by default */>); \
579 MatMulOp<GPUDevice, T, true /* cublas */>)
637 MatMulOp<SYCLDevice, T, false /* xxblas */>); \
642 MatMulOp<SYCLDevice, T, false /* xxblas */>)
/external/tensorflow/tensorflow/c/eager/
Dc_api_test_util.h44 TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b);
Dc_api_test.cc49 TFE_Op* matmul = MatMulOp(ctx, m, m); in BM_InitOp()
71 TFE_Op* matmul = MatMulOp(ctx, m, m); in BM_Execute()
170 TFE_Op* matmul = MatMulOp(ctx, h0_task1, h1_task1); in TestRemoteExecute()
252 TFE_Op* matmul = MatMulOp(ctx, h0_task0, h1_task2); in TestRemoteExecuteSilentCopies()
323 TFE_Op* matmul = MatMulOp(ctx, h0_task0, h0_task0); in CheckRemoteMatMulExecutesOK()
673 TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu); in TensorHandleSilentCopy()
719 TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu); in TensorHandleSilentCopyLocal()
750 TFE_Op* matmul = MatMulOp(ctx, m, m); in SetAndGetOpDevices()
882 TFE_Op* matmul = MatMulOp(ctx, m, m); in Execute_MatMul_CPU()
919 TFE_Op* matmul = MatMulOp(ctx, m1, m2); in Execute_MatMul_CPU_Runtime_Error()
[all …]
Dc_api_experimental_test.cc52 TFE_Op* matmul = MatMulOp(ctx, m, m); in ExecuteWithProfiling()
Dc_api_test_util.cc116 TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b) { in MatMulOp() function
/external/tensorflow/tensorflow/c/
Dc_api_experimental_test.cc188 TFE_Op* matmul_op = MatMulOp(ctx, m, m); in TEST()
367 TFE_Op* op = MatMulOp(eager_ctx_, m, m); in TEST_F()