Home
last modified time | relevance | path

Searched defs:MatMul (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op.h43 void MatMul(const Device& d, Out out, In0 in0, In1 in1, in MatMul() function
/external/llvm-project/llvm/lib/Transforms/Scalar/
DLowerMatrixIntrinsics.cpp1091 CallInst *MatMul) { in getNonAliasingPointer()
1170 bool isFusionProfitable(CallInst *MatMul) { in isFusionProfitable()
1212 void createTiledLoops(CallInst *MatMul, Value *LPtr, ShapeInfo LShape, in createTiledLoops()
1269 void emitSIMDTiling(CallInst *MatMul, LoadInst *LoadOp0, LoadInst *LoadOp1, in emitSIMDTiling()
1342 void LowerMatrixMultiplyFused(CallInst *MatMul, in LowerMatrixMultiplyFused()
1367 void LowerMultiply(CallInst *MatMul) { in LowerMultiply()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_matmul.cc33 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, in MatMul() function
Druntime_matmul.cc37 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, in MatMul() function
/external/tensorflow/tensorflow/c/experimental/ops/
Dmath_ops.cc89 Status MatMul(AbstractContext* ctx, in MatMul() function
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/Scalar/
DLowerMatrixIntrinsics.cpp703 void LowerMultiply(CallInst *MatMul) { in LowerMultiply()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner_test.cc135 TEST_F(ShapeRefinerTest, MatMul) { in TEST_F() argument
/external/tensorflow/tensorflow/core/ops/
Dmath_grad_test.cc272 Tensor MatMul(const Tensor& x, bool ax, const Tensor& y, bool ay) { in MatMul() function in tensorflow::__anoncd3cd74f0111::MathGradTest
/external/tensorflow/tensorflow/c/
Dc_api_test.cc1767 TF_Operation* MatMul(TF_Graph* graph, TF_Status* s, TF_Operation* l, in MatMul() function in tensorflow::__anon4051605d0111::CApiGradientsTest
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc2239 TEST_F(OpTest, MatMul) { in TEST_F() argument
/external/deqp/modules/glshared/
DglsBuiltinPrecisionTests.cpp3611 class MatMul : public MulFunc<Matrix<float, LeftRows, RightCols>, class
/external/deqp/external/vulkancts/modules/vulkan/shaderexecutor/
DvktShaderBuiltinPrecisionTests.cpp4608 class MatMul : public MulFunc<Matrix<T, LeftRows, RightCols>, class
/external/tensorflow/tensorflow/go/op/
Dwrappers.go32498 func MatMul(scope *Scope, a tf.Output, b tf.Output, optional ...MatMulAttr) (product tf.Output) { func