| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matmul_op.cc | 30 class MatMulOp : public XlaOpKernel { class
32 explicit MatMulOp(OpKernelConstruction* ctx, bool is_sparse = false) in MatMulOp() function in tensorflow::__anon7b49604f0111::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/ |
| D | matmul_op.cc | 442 class MatMulOp : public OpKernel { class
444 explicit MatMulOp(OpKernelConstruction* ctx) in MatMulOp() function in tensorflow::MatMulOp
566 MatMulOp<CPUDevice, T, false /* cublas, ignored for CPU */>);
571 MatMulOp<CPUDevice, T, false /* cublas, ignored for CPU */>); \
577 MatMulOp<GPUDevice, T, true /* cublas, true by default */>); \
582 MatMulOp<GPUDevice, T, true /* cublas */>)
605 MatMulOp<SYCLDevice, T, false /* xxblas */>); \
610 MatMulOp<SYCLDevice, T, false /* xxblas */>)
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_remote_test.cc | 87 TFE_Op* matmul = MatMulOp(ctx, h0_task1, h1_task1); in TestRemoteExecute()
171 TFE_Op* matmul = MatMulOp(ctx, h0_task0, h1_task2); in TestRemoteExecuteSilentCopies()
260 TFE_Op* matmul = MatMulOp(ctx, h0_task1, h1_task1); in TestRemoteExecuteDeleteContextWithOutstandingRPC()
315 TFE_Op* matmul = MatMulOp(ctx, h0_task0, h0_task0); in CheckRemoteMatMulExecutesOK()
|
| D | c_api_test_util.h | 46 TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b);
|
| D | c_api_test.cc | 50 TFE_Op* matmul = MatMulOp(ctx, m, m); in BM_InitOp()
72 TFE_Op* matmul = MatMulOp(ctx, m, m); in BM_Execute()
387 TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu); in TensorHandleSilentCopy()
435 TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu); in TensorHandleSilentCopyLocal()
468 TFE_Op* matmul = MatMulOp(ctx, m, m); in SetAndGetOpDevices()
602 TFE_Op* matmul = MatMulOp(ctx, m, m); in Execute_MatMul_CPU()
639 TFE_Op* matmul = MatMulOp(ctx, m1, m2); in Execute_MatMul_CPU_Runtime_Error()
643 TFE_Op* matmul2 = MatMulOp(ctx, m1, m1); in Execute_MatMul_CPU_Runtime_Error()
711 TFE_Op* matmul = MatMulOp(ctx, m1, m2); in Execute_MatMul_CPU_Type_Error()
775 TFE_Op* matmul = MatMulOp(ctx, m, m); in Execute_MatMul_XLA_CPU()
[all …]
|
| D | c_api_experimental_test.cc | 52 TFE_Op* matmul = MatMulOp(ctx, m, m); in ExecuteWithProfiling()
465 TFE_Op* matmul = MatMulOp(ctx, m, m); in Executor_MatMul_CPU()
|
| D | c_api_test_util.cc | 134 TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b) { in MatMulOp() function
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | unroll_batch_matmul.cc | 177 auto matmul = rewriter.create<TF::MatMulOp>(loc, matmul_type, in createMatMulOps()
258 rewriter.replaceOpWithNewOp<TF::MatMulOp>(op, result_type, in matchAndRewrite()
|
| D | legalize_tf.cc | 166 auto tf_matmul_op = cast<TF::MatMulOp>(op); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 96 TFE_Op* matmul_op = MatMulOp(ctx, m, m); in TEST()
|