Home
last modified time | relevance | path

Searched defs:matmul (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc49 TFE_Op* matmul = MatMulOp(ctx, m, m); in BM_InitOp() local
71 TFE_Op* matmul = MatMulOp(ctx, m, m); in BM_Execute() local
170 TFE_Op* matmul = MatMulOp(ctx, h0_task1, h1_task1); in TestRemoteExecute() local
252 TFE_Op* matmul = MatMulOp(ctx, h0_task0, h1_task2); in TestRemoteExecuteSilentCopies() local
323 TFE_Op* matmul = MatMulOp(ctx, h0_task0, h0_task0); in CheckRemoteMatMulExecutesOK() local
673 TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu); in TensorHandleSilentCopy() local
719 TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu); in TensorHandleSilentCopyLocal() local
750 TFE_Op* matmul = MatMulOp(ctx, m, m); in SetAndGetOpDevices() local
882 TFE_Op* matmul = MatMulOp(ctx, m, m); in Execute_MatMul_CPU() local
919 TFE_Op* matmul = MatMulOp(ctx, m1, m2); in Execute_MatMul_CPU_Runtime_Error() local
[all …]
Dc_api_experimental_test.cc52 TFE_Op* matmul = MatMulOp(ctx, m, m); in ExecuteWithProfiling() local
/external/tensorflow/tensorflow/core/grappler/costs/
Danalytical_cost_estimator_test.cc79 auto matmul = ops::MatMul(s.WithOpName("matmul"), flat, w2); in CreateMiniGraph() local
/external/tensorflow/tensorflow/cc/profiler/
Dprofiler_test.cc137 const GraphNodeProto* matmul = ExtractNode(ret, "y"); in TEST_F() local
/external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/
DTestUtil.java68 public static <T> Output<T> matmul( in matmul() method in TestUtil
/external/tensorflow/tensorflow/core/kernels/
Deigen_mkldnn_contraction_kernel_test.cc132 Scalar matmul = matmul_result(i % m, i / m); in TEST() local
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_algebra.py116 def matmul(lin_op_a, lin_op_b, name=None): function
Dlinear_operator.py570 def matmul(self, x, adjoint=False, adjoint_arg=False, name="matmul"): member in LinearOperator
/external/tensorflow/tensorflow/core/graph/
Dquantize_training_test.cc296 Node* matmul = test::graph::Matmul(graph, relu, identity, false, false); in TEST_F() local
333 Node* matmul = test::graph::Matmul(graph, relu, identity, false, false); in TEST_F() local
Dmkl_layout_pass.cc705 string matmul; member
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph_test.cc69 auto matmul = ops::MatMul(s.WithOpName("matmul"), feed, const_1); in TEST() local
Dconvert_nodes_test.cc1593 auto matmul = in TEST_F() local
/external/tensorflow/tensorflow/compiler/aot/tests/
Dtfcompile_test.cc227 foo::bar::MatMulComp matmul; in TEST() local
298 foo::bar::MatMulComp matmul( in TEST() local
/external/tensorflow/tensorflow/contrib/labeled_tensor/
D__init__.py113 matmul = _ops.matmul variable
/external/tensorflow/tensorflow/python/eager/
Dfunction_test.py1882 def matmul(x, y): function
1935 def matmul(x, y): function
2052 def matmul(x, y): function
2075 def matmul(x, y): function
/external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/
Dops.py765 def matmul(a, b, name=None): function
/external/tensorflow/tensorflow/python/kernel_tests/
Dpy_func_test.py50 def matmul(x, y): function
/external/python/cpython3/Lib/
Doperator.py108 def matmul(a, b): function
/external/tensorflow/tensorflow/python/ops/
Dinit_ops.py1100 def matmul(p1, p2, p3): function
Dmath_ops.py2403 def matmul(a, function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer_test.cc802 Output matmul = ops::MatMul(matmul_op, trans_a, trans_b); in TEST_F() local
804 Output matmul = ops::SparseMatMul(matmul_op, trans_a, trans_b); in TEST_F() local
806 Output matmul = ops::BatchMatMul(matmul_op, trans_a, trans_b); in TEST_F() local
862 Output matmul = ops::BatchMatMul(s.WithOpName("matmul"), trans_a, trans_b); in TEST_F() local
Darithmetic_optimizer.cc2177 const NodeScopeAndName matmul = ParseNodeScopeAndName(node->name()); in TrySimplify() local
2283 const NodeScopeAndName matmul = ParseNodeScopeAndName(node->name()); in TrySimplify() local
/external/tensorflow/tensorflow/c/
Dc_api_test.cc1651 TF_Operation* matmul = MatMul(graph_, s_, const0, const1, "MatMul"); in BuildSuccessGraph() local
1691 TF_Operation* matmul = in BuildExpectedGraph() local
/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc1123 auto* matmul = new TensorFlowMatMulOperator; in ConvertMatMulOperator() local