Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc131 TFE_Op* matmul = MatMulOp(ctx, m, m); in BM_InitOp() local
151 TFE_Op* matmul = MatMulOp(ctx, m, m); in BM_Execute() local
355 TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu); in TEST() local
396 TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu); in TEST() local
422 TFE_Op* matmul = MatMulOp(ctx, m, m); in TEST() local
453 TFE_Op* matmul = MatMulOp(ctx, m, m); in TEST() local
520 TFE_Op* matmul = MatMulOp(ctx, m, m); in TEST() local
598 TFE_Op* matmul = MatMulOp(ctx, m, m); in TEST() local
845 TFE_Op* matmul = TFE_NewOp(ctx, "MatMulFunction", status); in BM_ExecuteFunction() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfake_quantize_training_test.cc50 Output matmul = MatMul(root.WithOpName("matmul"), a_const, b_const); in TEST_F() local
Dremove_ema_test.cc55 Output matmul = MatMul(root.WithOpName("matmul"), a_const, b_const); in TEST_F() local
/external/tensorflow/tensorflow/core/grappler/costs/
Danalytical_cost_estimator_test.cc80 auto matmul = ops::MatMul(s.WithOpName("matmul"), flat, w2); in CreateMiniGraph() local
/external/tensorflow/tensorflow/compiler/aot/tests/
Dtfcompile_test.cc189 foo::bar::MatMulComp matmul; in TEST() local
260 foo::bar::MatMulComp matmul( in TEST() 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.java43 public static <T> Output<T> matmul( in matmul() method in TestUtil
/external/tensorflow/tensorflow/core/graph/
Dquantize_training_test.cc295 Node* matmul = test::graph::Matmul(graph, relu, identity, false, false); in TEST_F() local
332 Node* matmul = test::graph::Matmul(graph, relu, identity, false, false); in TEST_F() local
Dmkl_layout_pass.cc473 string matmul; member
/external/tensorflow/tensorflow/contrib/labeled_tensor/
D__init__.py113 matmul = _ops.matmul variable
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer_test.cc542 Output matmul = ops::MatMul(s.WithOpName("matmul"), trans_a, trans_b); in TEST_F() local
544 Output matmul = in TEST_F() local
547 Output matmul = in TEST_F() local
591 Output matmul = ops::BatchMatMul(s.WithOpName("matmul"), trans_a, trans_b); in TEST_F() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dpy_func_test.py44 def matmul(x, y): function
/external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/
Dops.py763 def matmul(a, b, name=None): function
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator.py586 def matmul(self, x, adjoint=False, adjoint_arg=False, name="matmul"): member in LinearOperator
/external/python/cpython3/Lib/
Doperator.py108 def matmul(a, b): function
/external/tensorflow/tensorflow/python/ops/
Dmath_ops.py1898 def matmul(a, function
/external/tensorflow/tensorflow/c/
Dc_api_test.cc1570 TF_Operation* matmul = MatMul(graph_, s_, const0, const1, "MatMul"); in BuildSuccessGraph() local
1610 TF_Operation* matmul = in BuildExpectedGraph() local
/external/tensorflow/tensorflow/contrib/lite/toco/
Dimport_tensorflow.cc1008 auto* matmul = new TensorFlowMatMulOperator; in ConvertMatMulOperator() local