Home
last modified time | relevance | path

Searched defs:matmul_op (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_tensorflow_matmul.cc66 const auto* matmul_op = in Run() local
Dunroll_batch_matmul.cc108 auto* matmul_op = new TensorFlowMatMulOperator; in UnrollBatchMatMul3D() local
265 auto* matmul_op = new TensorFlowMatMulOperator; in Run() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_batch_norms_test.cc215 Output matmul_op = in TestFoldBatchNormsMatMul() local
/external/tensorflow/tensorflow/c/
Dc_api_experimental_test.cc188 TFE_Op* matmul_op = MatMulOp(ctx, m, m); in TEST() local
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc639 tensorflow::NodeDef* matmul_op = tensorflow_graph->add_node(); in ConvertFullyConnectedOperator() local
1495 tensorflow::NodeDef* matmul_op = tensorflow_graph->add_node(); in ConvertLstmCellOperator() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer_test.cc800 auto matmul_op = s.WithOpName("matmul"); in TEST_F() local