| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_tensorflow_matmul.cc | 66 const auto* matmul_op = in Run() local
74 if (matmul_op->transpose_a) { in Run()
78 LogName(*matmul_op)); in Run()
86 string input_lhs = matmul_op->inputs[0]; in Run()
87 string input_rhs = matmul_op->inputs[1]; in Run()
88 if (!matmul_op->transpose_b) { in Run()
97 LogName(*matmul_op), input_rhs); in Run()
115 LogName(*matmul_op), input_rhs); in Run()
124 if (matmul_it->get() == matmul_op) { in Run()
128 DCHECK_EQ(matmul_it->get(), matmul_op); in Run()
[all …]
|
| D | unroll_batch_matmul.cc | 108 auto* matmul_op = new TensorFlowMatMulOperator; in UnrollBatchMatMul3D() local
109 matmul_op->inputs = {slice_a_reshape_op->outputs[0], in UnrollBatchMatMul3D()
111 matmul_op->outputs = {AvailableArrayName(*model, batch_name)}; in UnrollBatchMatMul3D()
112 auto& matmul_op_output = model->GetOrCreateArray(matmul_op->outputs[0]); in UnrollBatchMatMul3D()
114 *tail_it = model->operators.emplace(*tail_it, matmul_op) + 1; in UnrollBatchMatMul3D()
117 pack_inputs->push_back(matmul_op->outputs[0]); in UnrollBatchMatMul3D()
265 auto* matmul_op = new TensorFlowMatMulOperator; in Run() local
266 matmul_op->inputs = {input_lhs, input_rhs}; in Run()
267 matmul_op->outputs = batch_op->outputs; in Run()
268 tail_it = model->operators.emplace(tail_it, matmul_op) + 1; in Run()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_batch_norms_test.cc | 215 Output matmul_op = in TestFoldBatchNormsMatMul() local
223 Output mul_op = Mul(root.WithOpName("output"), matmul_op, mul_values_op); in TestFoldBatchNormsMatMul()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 639 tensorflow::NodeDef* matmul_op = tensorflow_graph->add_node(); in ConvertFullyConnectedOperator() local
640 matmul_op->set_op("MatMul"); in ConvertFullyConnectedOperator()
641 matmul_op->set_name(matmul_output); in ConvertFullyConnectedOperator()
642 *matmul_op->add_input() = reshape_output; in ConvertFullyConnectedOperator()
643 *matmul_op->add_input() = transpose_op->name(); in ConvertFullyConnectedOperator()
644 (*matmul_op->mutable_attr())["T"].set_type( in ConvertFullyConnectedOperator()
646 (*matmul_op->mutable_attr())["transpose_a"].set_b(false); in ConvertFullyConnectedOperator()
647 (*matmul_op->mutable_attr())["transpose_b"].set_b(false); in ConvertFullyConnectedOperator()
1495 tensorflow::NodeDef* matmul_op = tensorflow_graph->add_node(); in ConvertLstmCellOperator() local
1496 matmul_op->set_op("MatMul"); in ConvertLstmCellOperator()
[all …]
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 188 TFE_Op* matmul_op = MatMulOp(ctx, m, m); in TEST() local
194 TFE_ExecuteOpInNewThread(matmul_op, &retvals[0], &num_retvals, status); in TEST()
210 TFE_DeleteOp(matmul_op); in TEST()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | BUILD | 366 "//tensorflow/core/kernels:matmul_op",
507 "//tensorflow/core/kernels:matmul_op",
|
| /external/tensorflow/tensorflow/contrib/cmake/ |
| D | tf_core_kernels.cmake | 32 "${tensorflow_source_dir}/tensorflow/core/kernels/matmul_op.h"
33 "${tensorflow_source_dir}/tensorflow/core/kernels/matmul_op.cc"
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | BUILD | 200 "//tensorflow/core/kernels:matmul_op",
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | BUILD | 242 "//tensorflow/core/kernels:matmul_op",
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 54 "matmul_op.cc",
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 3269 ":matmul_op",
3363 name = "matmul_op",
3365 "matmul_op.cc",
3369 hdrs = ["matmul_op.h"],
3557 ":matmul_op",
3660 ":matmul_op",
5514 "matmul_op.cc",
5515 "matmul_op.h",
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 4405 "//tensorflow/core/kernels:matmul_op",
4551 "//tensorflow/core/kernels:matmul_op",
4593 "//tensorflow/core/kernels:matmul_op",
4627 "//tensorflow/core/kernels:matmul_op",
4718 "//tensorflow/core/kernels:matmul_op",
4754 "//tensorflow/core/kernels:matmul_op",
4846 "//tensorflow/core/kernels:matmul_op",
|
| /external/tensorflow/tensorflow/contrib/makefile/ |
| D | tf_op_files.txt | 151 tensorflow/core/kernels/matmul_op.cc
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 800 auto matmul_op = s.WithOpName("matmul"); in TEST_F() local
802 Output matmul = ops::MatMul(matmul_op, trans_a, trans_b); in TEST_F()
804 Output matmul = ops::SparseMatMul(matmul_op, trans_a, trans_b); in TEST_F()
806 Output matmul = ops::BatchMatMul(matmul_op, trans_a, trans_b); in TEST_F()
|