| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| resolve_tensorflow_matmul.cc | 66 const auto* matmul_op = local
74 if (matmul_op->transpose_a) {
78 LogName(*matmul_op));
86 string input_lhs = matmul_op->inputs[0];
87 string input_rhs = matmul_op->inputs[1];
88 if (!matmul_op->transpose_b) {
97 LogName(*matmul_op), input_rhs);
115 LogName(*matmul_op), input_rhs);
124 if (matmul_it->get() == matmul_op) {
128 DCHECK_EQ(matmul_it->get(), matmul_op);
[all...] |
| unroll_batch_matmul.cc | 108 auto* matmul_op = new TensorFlowMatMulOperator; local
109 matmul_op->inputs = {slice_a_reshape_op->outputs[0],
111 matmul_op->outputs = {AvailableArrayName(*model, batch_name)};
112 auto& matmul_op_output = model->GetOrCreateArray(matmul_op->outputs[0]);
114 *tail_it = model->operators.emplace(*tail_it, matmul_op) + 1;
117 pack_inputs->push_back(matmul_op->outputs[0]);
265 auto* matmul_op = new TensorFlowMatMulOperator; local
266 matmul_op->inputs = {input_lhs, input_rhs};
267 matmul_op->outputs = batch_op->outputs;
268 tail_it = model->operators.emplace(tail_it, matmul_op) + 1
[all...] |
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| fold_batch_norms_test.cc | 215 Output matmul_op = local
216 MatMul(root.WithOpName("matmul_op"), input_op, weights_op);
223 Output mul_op = Mul(root.WithOpName("output"), matmul_op, mul_values_op);
|
| /external/tensorflow/tensorflow/c/ |
| c_api_experimental_test.cc | 188 TFE_Op* matmul_op = MatMulOp(ctx, m, m); local
194 TFE_ExecuteOpInNewThread(matmul_op, &retvals[0], &num_retvals, status);
210 TFE_DeleteOp(matmul_op);
|
| /external/tensorflow/tensorflow/lite/toco/ |
| export_tensorflow.cc | 639 tensorflow::NodeDef* matmul_op = tensorflow_graph->add_node(); local
640 matmul_op->set_op("MatMul");
641 matmul_op->set_name(matmul_output);
642 *matmul_op->add_input() = reshape_output;
643 *matmul_op->add_input() = transpose_op->name();
644 (*matmul_op->mutable_attr())["T"].set_type(
646 (*matmul_op->mutable_attr())["transpose_a"].set_b(false);
647 (*matmul_op->mutable_attr())["transpose_b"].set_b(false);
1495 tensorflow::NodeDef* matmul_op = tensorflow_graph->add_node(); local
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| arithmetic_optimizer_test.cc | 800 auto matmul_op = s.WithOpName("matmul"); local
802 Output matmul = ops::MatMul(matmul_op, trans_a, trans_b);
804 Output matmul = ops::SparseMatMul(matmul_op, trans_a, trans_b);
806 Output matmul = ops::BatchMatMul(matmul_op, trans_a, trans_b);
[all...] |
