Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dquantize_training_test.cc94 Node* m1 = test::graph::Matmul(g, relu, identity, false, false); in TEST_F()
145 Node* m1 = test::graph::Matmul(g, relu, relu6, false, false); in TEST_F()
194 Node* m1 = test::graph::Matmul(g, relu, identity, false, false); in TEST_F()
195 Node* m2 = test::graph::Matmul(g, identity, c, false, false); in TEST_F()
248 Node* m1 = test::graph::Matmul(g, relu, identity, false, false); in TEST_F()
249 Node* m2 = test::graph::Matmul(g, identity, c, false, false); in TEST_F()
296 Node* matmul = test::graph::Matmul(graph, relu, identity, false, false); in TEST_F()
333 Node* matmul = test::graph::Matmul(graph, relu, identity, false, false); in TEST_F()
375 Node* m1 = test::graph::Matmul(g, relu, relu6, false, false); in TEST_F()
469 Node* m1 = test::graph::Matmul(g, relu, relu6, false, false); in TEST_F()
Ddirect_session_with_tracking_alloc_test.cc62 Node* y = test::graph::Matmul(&graph, a, x, false, false); in TEST()
180 Node* y = test::graph::Matmul(&graph, a, x, false, false); in TestHWAccelerator()
262 Node* y = test::graph::Matmul(&graph, a, x, false, false); in TEST()
324 Node* y = test::graph::Matmul(&graph, a, x, false, false); in TEST()
Ddirect_session_test.cc110 Node* y = test::graph::Matmul(&graph, a, x, false, false); in Initialize()
687 Node* y = test::graph::Matmul(&graph, a, x, false, false); in TEST_F()
/external/llvm-project/mlir/include/mlir/Dialect/Vector/
DVectorOps.h54 Matmul = 1, enumerator
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_patterns.td45 // Matmul without transpose on b to matmul with explicit transpose op and
54 // Matmul with transpose on a to matmul with explicit transpose op and a not
/external/llvm-project/mlir/test/lib/Transforms/
DTestLinalgCodegenStrategy.cpp111 .Case("matrixintrinsics", vector::VectorContractLowering::Matmul) in runOnFunction()
DTestVectorTransforms.cpp113 contractLowering = VectorContractLowering::Matmul; in runOnFunction()
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op_test.cc350 static Graph* Matmul(int m, int k, int n, bool transpose_a, bool transpose_b, in Matmul() function
357 test::graph::Matmul(g, test::graph::Constant(g, in0), in Matmul()
367 test::Benchmark(#DEVICE, Matmul<T>(M, K, N, TA, TB, TFTYPE)).Run(iters); \
/external/llvm-project/mlir/integration_test/Dialect/Linalg/CPU/
Dmatmul-vs-matvec.mlir59 assert %c, "Matmul does not produce same output as matvec"
/external/tensorflow/tensorflow/core/graph/
Dtestlib.h81 Node* Matmul(Graph* g, Node* in0, Node* in1, bool transpose_a,
Dtestlib.cc159 Node* Matmul(Graph* g, Node* in0, Node* in1, bool transpose_a, in Matmul() function
/external/tensorflow/tensorflow/core/debug/
Dgrpc_session_debug_test.cc66 Node* c = test::graph::Matmul(&graph, a, b, false, false); in CreateGraphDef()
/external/tensorflow/tensorflow/python/framework/
Dfunction_test.py1854 def Matmul(a, b): function
1857 Matmul(1., 2.)
1870 def Matmul(a, b): function
1879 Divide(Matmul(1., 2.), 3.)
1901 def Matmul(a, b): function
1904 return Matmul(a, b)
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Ddevice_tracer_test.cc90 Node* y = test::graph::Matmul(&graph, a, x, false, false); in Initialize()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_test.cc379 Node* y_node = test::graph::Matmul(&graph, a_node, x_node, false, false); in TEST_F()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session_test.cc57 Node* c = test::graph::Matmul(&graph, a, b, false, false); in CreateGraphDef()
342 Node* y = test::graph::Matmul(&graph, a, x, false, false); in FindMaxEigen()
427 Node* c = test::graph::Matmul(&graph, a, b, false, false); in TEST()
/external/tensorflow/tensorflow/core/framework/
Drun_handler_test.cc614 Node* y = test::graph::Matmul(&graph, a, x, false, false); in Initialize()
/external/llvm-project/mlir/lib/Dialect/Vector/
DVectorTransforms.cpp1625 vector::VectorContractLowering::Matmul) in matchAndRewrite()
/external/llvm-project/mlir/test/Transforms/
Dloop-fusion.mlir2446 // Matmul.
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf.mlir1147 // Matmul op legalizations.