OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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