Home
last modified time | relevance | path

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

/aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/Transforms/Scalar/
H A DLowerMatrixIntrinsics.cpp1406 CallInst *MatMul) { in getNonAliasingPointer()
1491 bool isFusionProfitable(CallInst *MatMul) { in isFusionProfitable()
1535 void createTiledLoops(CallInst *MatMul, Value *LPtr, ShapeInfo LShape, in createTiledLoops()
1594 void emitSIMDTiling(CallInst *MatMul, LoadInst *LoadOp0, LoadInst *LoadOp1, in emitSIMDTiling()
1662 void LowerMatrixMultiplyFused(CallInst *MatMul, in LowerMatrixMultiplyFused()
1758 void LowerMultiply(CallInst *MatMul) { in LowerMultiply()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmatmul_op.h43 void MatMul(const Device& d, Out out, In0 in0, In1 in1, in MatMul() function
/aosp_15_r20/external/executorch/backends/arm/test/ops/
H A Dtest_bmm.py35 class MatMul(torch.nn.Module): class in TestBMM
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_single_threaded_matmul.cc32 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, int64_t m, in MatMul() function
H A Druntime_matmul.cc36 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, int64_t m, in MatMul() function
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/ops/
H A Dmath_ops.cc108 Status MatMul(AbstractContext* ctx, AbstractTensorHandle* const a, in MatMul() function
/aosp_15_r20/external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/Scalar/
H A DLowerMatrixIntrinsics.cpp703 void LowerMultiply(CallInst *MatMul) { in LowerMultiply()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dshape_refiner_test.cc137 TEST_F(ShapeRefinerTest, MatMul) { in TEST_F() argument
/aosp_15_r20/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/
H A Donedevice_train_palm_detection_filter_inference_beta.cc185 void MatMul(const int32_t* __restrict input_shape, in MatMul() function
H A Donedevice_train_palm_detection_filter_inference.cc186 void MatMul(const int32_t* __restrict input_shape, in MatMul() function
H A Donedevice_train_palm_detection_filter_inference_v2.cc185 void MatMul(const int32_t* __restrict input_shape, in MatMul() function
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dmath_grad_test.cc272 Tensor MatMul(const Tensor& x, bool ax, const Tensor& y, bool ay) { in MatMul() function in tensorflow::__anone11a807e0111::MathGradTest
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api_test.cc1798 TF_Operation* MatMul(TF_Graph* graph, TF_Status* s, TF_Operation* l, in MatMul() function in tensorflow::__anonfb0de0ec0111::CApiGradientsTest
/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_static_runtime.cc840 TEST(StaticRuntime, MatMul) { in TEST() argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Drandomized_tests.cc3233 TEST_F(OpTest, MatMul) { in TEST_F() argument
/aosp_15_r20/external/deqp/modules/glshared/
H A DglsBuiltinPrecisionTests.cpp3856 class MatMul : public MulFunc<Matrix<float, LeftRows, RightCols>, Matrix<float, LeftRows, Middle>, class
/aosp_15_r20/external/deqp/external/vulkancts/modules/vulkan/shaderexecutor/
H A DvktShaderBuiltinPrecisionTests.cpp4993 class MatMul : public MulFunc<Matrix<T, LeftRows, RightCols>, Matrix<T, LeftRows, Middle>, Matrix<T… class
/aosp_15_r20/external/tensorflow/tensorflow/go/op/
H A Dwrappers.go23866 func MatMul(scope *Scope, a tf.Output, b tf.Output, optional ...MatMulAttr) (product tf.Output) { func