HomeSort by relevance Sort by last modified time
    Searched defs:BatchMatMul (Results 1 - 4 of 4) sorted by null

  /external/eigen/unsupported/test/
cxx11_tensor_custom_op.cpp 61 struct BatchMatMul {
91 Tensor<float, 3> result = tensor1.customOp(tensor2, BatchMatMul());
  /external/tensorflow/tensorflow/core/kernels/
batch_matmul_op_impl.h 441 class BatchMatMul : public OpKernel {
443 explicit BatchMatMul(OpKernelConstruction* context) : OpKernel(context) {
448 virtual ~BatchMatMul() {}
511 Name("BatchMatMul").Device(DEVICE_CPU).TypeConstraint<TYPE>("T"), \
512 BatchMatMul<CPUDevice, TYPE>)
516 Name("BatchMatMul").Device(DEVICE_GPU).TypeConstraint<TYPE>("T"), \
517 BatchMatMul<GPUDevice, TYPE>)
522 Name("BatchMatMul").Device(DEVICE_SYCL).TypeConstraint<TYPE>("T"), \
523 BatchMatMul<SYCLDevice, TYPE>)
  /external/tensorflow/tensorflow/core/ops/
math_grad_test.cc 235 Tensor BatchMatMul(const Tensor& x, bool ax, const Tensor& y, bool ay) {
236 return MatMulCommon("BatchMatMul", "adj_x", "adj_y", x, ax, y, ay);
307 return MatMulGradCommon("BatchMatMul", "adj_x", "adj_y", x, ax, y, ay, dx,
    [all...]
  /external/tensorflow/tensorflow/go/op/
wrappers.go     [all...]

Completed in 266 milliseconds