Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
Deigen_mkldnn_contraction_kernel_test.cc140 Scalar matmul = matmul_result(i % m, i / m); in TEST() local
211 Scalar matmul = matmul_result(i, j); in TEST() local
Dmatmul_op_test.cc106 ops::MatMul matmul = ops::MatMul( in RunMatMulWithBias() local
125 ops::MatMul matmul = ops::MatMul( in RunMatMulWithBiasAndActivation() local
207 Tensor matmul; in VerifyBiasAddTensorsNear() local
/external/tensorflow/tensorflow/c/eager/
Dcustom_device_test.cc49 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> matmul( in TEST() local
298 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> matmul( in TEST() local
Dc_api_test.cc63 TFE_Op* matmul = MatMulOp(ctx, m, m); in BM_InitOp() local
85 TFE_Op* matmul = TFE_NewOp(ctx, "MatMul", status); in BM_Execute() local
430 TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu); in TensorHandleSilentCopy() local
486 TFE_Op* matmul = MatMulOp(ctx, m, m); in SetAndGetOpDevices() local
728 TFE_Op* matmul = MatMulOp(ctx, m, m); in Execute_MatMul_CPU() local
765 TFE_Op* matmul = MatMulOp(ctx, m1, m2); in Execute_MatMul_CPU_Runtime_Error() local
837 TFE_Op* matmul = MatMulOp(ctx, m1, m2); in Execute_MatMul_CPU_Type_Error() local
901 TFE_Op* matmul = MatMulOp(ctx, m, m); in ExecuteWithTracing() local
1175 TFE_Op* matmul = TFE_NewOp(ctx, "MatMulFunction", status); in BM_ExecuteFunction() local
Dc_api_remote_test.cc77 TFE_Op* matmul = MatMulOp(ctx, h0_task1, h1_task1); in TestRemoteExecute() local
Dc_api_remote_test_util.cc137 TFE_Op* matmul = nullptr; in TestRemoteExecuteSilentCopies() local
Dc_api_distributed_test.cc593 TFE_Op* matmul = MatMulOp(ctx, h0_task1, h1_task1); in TestRemoteExecuteDeleteContextWithOutstandingRPC() local
Dc_api_cluster_test.cc63 TFE_Op* matmul = MatMulOp(ctx, h0_task0, h0_task0); in CheckRemoteMatMulExecutesOK() local
Dc_api_experimental_test.cc331 TFE_Op* matmul = MatMulOp(ctx, m, m); in Executor_MatMul_CPU() 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/compiler/mlir/tensorflow/transforms/
Dfused_kernel_matcher.cc222 bool AreFuseCompatible(MatMulOp matmul, BiasAddOp bias_add, in AreFuseCompatible()
Dunroll_batch_matmul.cc172 auto matmul = rewriter.create<TF::MatMulOp>(loc, matmul_type, in createMatMulOps() local
/external/tensorflow/tensorflow/core/common_runtime/
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
/external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/
DTestUtil.java74 public static <T> Output<T> matmul( in matmul() method in TestUtil
/external/python/cpython3/Lib/test/
Dtest_type_comments.py35 matmul = """\ variable
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_block_diag.py290 def matmul(self, x, adjoint=False, adjoint_arg=False, name="matmul"): member in LinearOperatorBlockDiag
Dlinear_operator_algebra.py122 def matmul(lin_op_a, lin_op_b, name=None): function
Dlinear_operator_block_lower_triangular.py394 def matmul(self, x, adjoint=False, adjoint_arg=False, name="matmul"): member in LinearOperatorBlockLowerTriangular
/external/tensorflow/tensorflow/compiler/aot/tests/
Dtfcompile_test.cc249 foo::bar::MatMulComp matmul; in TEST() local
320 foo::bar::MatMulComp matmul( in TEST() local
/external/tensorflow/tensorflow/python/ops/linalg/sparse/
Dsparse_csr_matrix_ops.py147 def matmul(a, function
/external/tensorflow/tensorflow/core/grappler/utils/
Dpattern_utils_test.cc42 auto matmul = ops::MatMul(s.WithOpName("matmul"), input, weight); in GetMatMulBiasAddGeluGraph() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmkl_remapper_test.cc466 auto matmul = ops::MatMul(s.WithOpName("matmul"), input, filter); in TEST_F() local
Dremapper_test.cc409 auto matmul = ops::MatMul(s.WithOpName("matmul"), lhs, rhs); in RunTest() local
658 auto matmul = ops::MatMul(s.WithOpName("matmul"), lhs, rhs); in RunTest() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dpy_func_test.py52 def matmul(x, y): function

12