Searched refs:output_op (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_hardswish.cc | 60 const auto* output_op = GetOpWithInput(*model, mul_op->outputs[0]); in Run() local
61 ops.push_back(output_op); in Run()
62 if (output_op->type == OperatorType::kFakeQuant) { in Run()
63 output_op = GetOpWithInput(*model, output_op->outputs[0]); in Run()
64 ops.push_back(output_op); in Run()
66 if (!IsBinaryOp(output_op, OperatorType::kMul)) { in Run()
82 mul_inputs.insert(mul_inputs.end(), output_op->inputs.begin(), in Run()
83 output_op->inputs.end()); in Run()
101 const auto output_tensor_name = output_op->outputs[0]; in Run()
|
| D | group_bidirectional_sequence_ops.cc | 83 bool FindUnidirectionalSequenceOp(const Model& model, const Operator& output_op, in FindUnidirectionalSequenceOp() argument
88 op_it = GetOpWithOutput(model, output_op.inputs[0]); in FindUnidirectionalSequenceOp()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | op_builder.cc | 312 auto* output_op = AddNode(); in AddOutputTensors() local
313 output_op->SetOpType(OP_OUTPUT); in AddOutputTensors()
315 output_op->AddInput(hexagon_output); in AddOutputTensors()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 1439 TF_Operation* output_op = in TEST() local
1441 ASSERT_TRUE(output_op != nullptr); in TEST()
1442 csession.SetOutputs({output_op}); in TEST()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | function_test.py | 901 output_op = Foo(input_op)
910 output_op, feed_dict={input_op: np.random.rand(32, 100)})
|