Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_batch_matmul_op.cc51 class BatchMatMulMkl : public OpKernel { class
53 explicit BatchMatMulMkl(OpKernelConstruction* context) in BatchMatMulMkl() function in tensorflow::BatchMatMulMkl
59 virtual ~BatchMatMulMkl() {} in ~BatchMatMulMkl()
236 BatchMatMulMkl<CPUDevice, TYPE, false>)
243 BatchMatMulMkl<CPUDevice, TYPE, true>)