| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_tensorflow_matmul.cc | 66 const auto* matmul_op = in Run() local
69 auto refresh_matmul_iterator = [&model, &matmul_it, &matmul_op]() { in Run()
71 [matmul_op](const std::unique_ptr<Operator>& op) { in Run()
72 return op.get() == matmul_op; in Run()
74 DCHECK_EQ(matmul_it->get(), matmul_op); in Run()
77 std::string input_lhs = matmul_op->inputs[0]; in Run()
78 std::string input_rhs = matmul_op->inputs[1]; in Run()
82 if (matmul_op->transpose_a) { in Run()
88 LogName(*matmul_op)); in Run()
132 if (!matmul_op->transpose_b) { in Run()
[all …]
|
| D | unroll_batch_matmul.cc | 189 auto* matmul_op = new TensorFlowMatMulOperator; in Run() local
190 matmul_op->inputs = {input_lhs, input_rhs}; in Run()
191 matmul_op->outputs = batch_op->outputs; in Run()
192 model->operators.emplace(tail_it, matmul_op); in Run()
221 auto* matmul_op = new TensorFlowMatMulOperator; in Run() local
222 matmul_op->inputs = {slice_a_outputs[a_batch_idx], in Run()
224 matmul_op->outputs = {AvailableArrayName(*model, batch_name)}; in Run()
225 auto& matmul_op_output = model->GetOrCreateArray(matmul_op->outputs[0]); in Run()
227 tail_it = model->operators.emplace(tail_it, matmul_op) + 1; in Run()
230 pack_inputs.push_back(matmul_op->outputs[0]); in Run()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | matmul_builder.cc | 50 OpBuilder* matmul_op, in AddFullyConnectedHelper() argument
65 matmul_op->AddInput(graph_builder->GetHexagonTensorId(data_tensor_id)); in AddFullyConnectedHelper()
66 matmul_op->AddInput(weights_id); in AddFullyConnectedHelper()
67 matmul_op->AddInput(OpBuilder::TensorID(data_min_const->GetID(), 0)); in AddFullyConnectedHelper()
68 matmul_op->AddInput(OpBuilder::TensorID(data_max_const->GetID(), 0)); in AddFullyConnectedHelper()
69 matmul_op->AddInput(weights_min_id); in AddFullyConnectedHelper()
70 matmul_op->AddInput(weights_max_id); in AddFullyConnectedHelper()
79 matmul_op->AddOutput(sizeof(int), 4, in AddFullyConnectedHelper()
83 matmul_op->AddOutput(sizeof(float), 4, scalar_shape); in AddFullyConnectedHelper()
85 matmul_op->AddOutput(sizeof(float), 4, scalar_shape); in AddFullyConnectedHelper()
[all …]
|
| D | matmul_builder.h | 65 TfLiteContext* context, OpBuilder* matmul_op);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | fold_broadcast.cc | 78 auto matmul_op = llvm::dyn_cast<TF::BatchMatMulV2Op>(op); in RewriteBatchMatMulV2Op() local
79 if (!matmul_op) return failure(); in RewriteBatchMatMulV2Op()
100 matmul_op.adj_x() ? shape_x.back() : *(shape_x.rbegin() + 1); in RewriteBatchMatMulV2Op()
102 !matmul_op.adj_x() ? shape_x.back() : *(shape_x.rbegin() + 1); in RewriteBatchMatMulV2Op()
105 matmul_op.adj_y() ? shape_y.back() : *(shape_y.rbegin() + 1); in RewriteBatchMatMulV2Op()
107 !matmul_op.adj_y() ? shape_y.back() : *(shape_y.rbegin() + 1); in RewriteBatchMatMulV2Op()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 165 TFE_Op* matmul_op; in TEST_F() local
166 matmul_op = TFE_NewOp(tfe_context_, "MatMul", status_); in TEST_F()
170 CheckOutputShapes(matmul_op, in TEST_F()
176 CheckOutputShapes(matmul_op, in TEST_F()
183 matmul_op, in TEST_F()
189 CheckOutputShapes(matmul_op, in TEST_F()
194 TFE_DeleteOp(matmul_op); in TEST_F()
|
| /external/tensorflow/tensorflow/c/experimental/ops/ |
| D | math_ops.cc | 93 AbstractOperationPtr matmul_op(ctx->CreateOperation()); in MatMul() local
94 TF_RETURN_IF_ERROR(matmul_op->Reset("MatMul", /*raw_device_name=*/nullptr)); in MatMul()
95 TF_RETURN_IF_ERROR(MaybeSetOpName(matmul_op.get(), name)); in MatMul()
96 TF_RETURN_IF_ERROR(matmul_op->AddInput(inputs[0])); in MatMul()
97 TF_RETURN_IF_ERROR(matmul_op->AddInput(inputs[1])); in MatMul()
99 TF_RETURN_IF_ERROR(matmul_op->SetAttrBool("transpose_a", transpose_a)); in MatMul()
100 TF_RETURN_IF_ERROR(matmul_op->SetAttrBool("transpose_b", transpose_b)); in MatMul()
103 TF_RETURN_IF_ERROR(matmul_op->Execute(outputs, &num_retvals)); in MatMul()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | BUILD | 66 "//tensorflow/core/kernels:matmul_op",
189 "//tensorflow/core/kernels:matmul_op",
428 "//tensorflow/core/kernels:matmul_op",
|
| /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 | 644 tensorflow::NodeDef* matmul_op = tensorflow_graph->add_node(); in ConvertFullyConnectedOperator() local
645 matmul_op->set_op("MatMul"); in ConvertFullyConnectedOperator()
646 matmul_op->set_name(matmul_output); in ConvertFullyConnectedOperator()
647 *matmul_op->add_input() = reshape_output; in ConvertFullyConnectedOperator()
648 *matmul_op->add_input() = transpose_op->name(); in ConvertFullyConnectedOperator()
649 (*matmul_op->mutable_attr())["T"].set_type( in ConvertFullyConnectedOperator()
651 (*matmul_op->mutable_attr())["transpose_a"].set_b(false); in ConvertFullyConnectedOperator()
652 (*matmul_op->mutable_attr())["transpose_b"].set_b(false); in ConvertFullyConnectedOperator()
1525 tensorflow::NodeDef* matmul_op = tensorflow_graph->add_node(); in ConvertLstmCellOperator() local
1526 matmul_op->set_op("MatMul"); in ConvertLstmCellOperator()
[all …]
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_unified_experimental_test.cc | 457 auto* matmul_op = TF_NewAbstractOp(graph_ctx); in TEST_P() local
458 TF_AbstractOpSetOpType(matmul_op, "MatMul", status.get()); in TEST_P()
460 TF_AbstractOpSetOpName(matmul_op, "my_matmul", status.get()); in TEST_P()
470 TF_ExecuteOperation(matmul_op, 2, inputs, mm_outputs, status.get()); in TEST_P()
474 TF_DeleteAbstractOp(matmul_op); in TEST_P()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | BUILD | 403 "//tensorflow/core/kernels:matmul_op",
561 "//tensorflow/core/kernels:matmul_op",
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | BUILD | 245 "//tensorflow/core/kernels:matmul_op",
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | BUILD | 2088 "//tensorflow/core/kernels:matmul_op",
2243 "//tensorflow/core/kernels:matmul_op",
2287 "//tensorflow/core/kernels:matmul_op",
2321 "//tensorflow/core/kernels:matmul_op",
2419 "//tensorflow/core/kernels:matmul_op",
2456 "//tensorflow/core/kernels:matmul_op",
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 63 "matmul_op.cc",
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | BUILD | 487 "//tensorflow/core/kernels:matmul_op",
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 3267 ":matmul_op",
3293 deps = [":matmul_op"],
3297 name = "matmul_op",
3304 prefix = "matmul_op",
3553 ":matmul_op",
3644 ":matmul_op",
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 727 auto matmul_op = s.WithOpName("matmul"); in TEST_F() local
729 matmul = ops::MatMul(matmul_op, trans_a, trans_b); in TEST_F()
731 matmul = ops::SparseMatMul(matmul_op, trans_a, trans_b); in TEST_F()
733 matmul = ops::BatchMatMul(matmul_op, trans_a, trans_b); in TEST_F()
735 matmul = ops::BatchMatMulV2(matmul_op, trans_a, trans_b); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | BUILD | 1108 "//tensorflow/core/kernels:matmul_op",
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | BUILD | 1137 "//tensorflow/core/kernels:matmul_op",
|