Home
last modified time | relevance | path

Searched refs:outputs_ (Results 1 – 25 of 50) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dsplit_test.cc28 outputs_.push_back(AddOutput(output)); in SplitOpModel()
42 return Dequantize<T>(ExtractVector<T>(outputs_[idx]), in GetDequantizedOutput()
43 GetScale(outputs_[idx]), GetZeroPoint(outputs_[idx])); in GetDequantizedOutput()
46 std::vector<int> GetOutputShape(int i) { return GetTensorShape(outputs_[i]); } in GetOutputShape()
51 std::vector<int> outputs_; member in tflite::SplitOpModel
/external/tensorflow/tensorflow/core/grappler/
Dutils.h114 outputs_.reserve(graph->node_size()); in NodeMapInternal()
126 outputs_[NodeName(input)].insert(canonical); in NodeMapInternal()
135 auto it = outputs_.find(node_name); in GetOutputs()
136 if (it == outputs_.end()) { in GetOutputs()
146 auto it = outputs_.find(node_name); in GetOutputsOrderedByNodeName()
147 if (it != outputs_.end()) { in GetOutputsOrderedByNodeName()
171 outputs_.erase(NodeName(name)); in RemoveNode()
193 outputs_[node_name].insert(output_node); in AddOutput()
197 outputs_[node_name].erase(nodes_[NodeName(output_name)]); in RemoveOutput()
213 void RemoveOutputs(const string& node_name) { outputs_.erase(node_name); } in RemoveOutputs()
[all …]
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dctc_beam_search_decoder_test.cc51 outputs_.push_back(AddOutput(TensorType_INT32)); in CTCBeamSearchDecoderOpModel()
53 outputs_.push_back(AddOutput(TensorType_FLOAT32)); in CTCBeamSearchDecoderOpModel()
73 for (int i = 0; i < outputs_.size() - 1; ++i) { in GetDecodedOutpus()
74 outputs.push_back(ExtractVector<int>(outputs_[i])); in GetDecodedOutpus()
80 return ExtractVector<float>(outputs_[outputs_.size() - 1]); in GetLogProbabilitiesOutput()
85 for (const int output : outputs_) { in GetOutputShapes()
94 std::vector<int> outputs_; member in tflite::ops::experimental::__anonc74d09040111::CTCBeamSearchDecoderOpModel
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc127 outputs_.push_back(index); in OpOutputs()
129 vector_.resize(outputs_.size()); in OpOutputs()
137 for (int i : outputs_) { in InitializeGraphOutputs()
160 int Size() const { return outputs_.size(); } in Size()
162 int TfLiteIndex(int i) const { return outputs_[i]; } in TfLiteIndex()
180 std::vector<int> outputs_; member in tflite::flex::OpOutputs
190 : inputs_(inputs), outputs_(outputs) {} in OpNode()
209 const OpOutputs& outputs() const { return outputs_; } in outputs()
210 OpOutputs* mutable_outputs() { return &outputs_; } in mutable_outputs()
213 int NumOutputs() const { return outputs_.Size(); } in NumOutputs()
[all …]
/external/tensorflow/tensorflow/c/
Dwhile_loop_test.cc57 outputs_.resize(ninputs, {nullptr, -1}); in Init()
61 TF_FinishWhile(params_.get(), s_, &outputs_[0]); in ExpectOK()
66 TF_FinishWhile(params_.get(), s_, &outputs_[0]); in ExpectError()
75 Run(outputs_, input_values); in Run()
128 std::vector<TF_Output> outputs_; // The final outputs of the while loop member in __anonbc36f9730111::CApiWhileLoopTest
177 EXPECT_TRUE(outputs_[0].oper != nullptr); in TEST_F()
178 EXPECT_GE(outputs_[0].index, 0); in TEST_F()
179 EXPECT_TRUE(outputs_[1].oper != nullptr); in TEST_F()
180 EXPECT_GE(outputs_[1].index, 0); in TEST_F()
433 TF_AddGradients(graph_, outputs_.data(), outputs_.size(), inputs_.data(), 1, in TEST_F()
Dc_test_util.cc481 outputs_.clear(); in SetOutputs()
483 outputs_.emplace_back(TF_Output{o, 0}); in SetOutputs()
485 output_values_.resize(outputs_.size()); in SetOutputs()
490 outputs_ = outputs; in SetOutputs()
491 output_values_.resize(outputs_.size()); in SetOutputs()
507 output_values_.resize(outputs_.size(), nullptr); in Run()
513 const TF_Output* outputs_ptr = outputs_.empty() ? nullptr : &outputs_[0]; in Run()
520 outputs_ptr, output_values_ptr, outputs_.size(), targets_ptr, in Run()
/external/tensorflow/tensorflow/cc/ops/
Dwhile_loop_test.cc43 ops::BuildWhileLoop(scope_, inputs_, cond, body, kFrameName, &outputs_); in CreateLoop()
60 TF_ASSERT_OK(session.Run(feeds, outputs_, &out_tensors)); in Run()
61 ASSERT_EQ(out_tensors.size(), outputs_.size()); in Run()
72 std::vector<Output> outputs_; member in tensorflow::__anon7e249d170111::WhileLoopTest
98 for (int i = 0; i < outputs_.size(); ++i) { in TEST_F()
99 Node* node = outputs_[i].node(); in TEST_F()
/external/tensorflow/tensorflow/cc/framework/
Dwhile_gradients_test.cc45 &outputs_)); in CreateLoop()
50 AddSymbolicGradients(scope_, outputs_, inputs_, &grad_outputs_)); in CreateBackprop()
87 std::vector<Output> outputs_; member in tensorflow::__anon97a6b1da0111::WhileGradientsTest
182 outputs_[0] = ops::Neg(scope_, outputs_[0]); in TEST_F()
213 AddSymbolicGradients(cpu1_scope, outputs_, inputs_, &grad_outputs_)); in TEST_F()
Dgradients.cc104 const std::vector<Output>& outputs_; member in tensorflow::__anonc8091ec30111::SymbolicGradientBuilder
147 outputs_(outputs), in SymbolicGradientBuilder()
171 for (const Output& out : outputs_) { in GetReachableNodes()
251 if (outputs_.size() != grad_inputs_.size()) { in Initialize()
268 output_nodes.reserve(outputs_.size()); in Initialize()
269 for (size_t i = 0; i < outputs_.size(); ++i) { in Initialize()
270 output_nodes.insert(outputs_[i].node()->id()); in Initialize()
323 for (const Output& output : outputs_) { in Initialize()
337 TF_RETURN_IF_ERROR(BackpropAlongEdge(grad_inputs_[i], outputs_[i])); in Initialize()
/external/tensorflow/tensorflow/lite/kernels/
Dsplit_v_test.cc45 outputs_.push_back(AddOutput(input.type)); in SplitVOpModel()
68 return ExtractVector<T>(outputs_[i]); in GetOutput()
70 std::vector<int> GetOutputShape(int i) { return GetTensorShape(outputs_[i]); } in GetOutputShape()
76 std::vector<int> outputs_; member in tflite::__anon3edbe8b00111::SplitVOpModel
Dunpack_test.cc42 outputs_.push_back(AddOutput(input.type)); in UnpackOpModel()
55 for (const int output : outputs_) { in GetOutputDatas()
64 for (const int output : outputs_) { in GetOutputShapes()
72 std::vector<int> outputs_; member in tflite::__anona011a1f10111::UnpackOpModel
Dsplit_test.cc50 outputs_.push_back(AddOutput(input.type)); in SplitOpModel()
69 return ExtractVector<T>(outputs_[i]); in GetOutput()
71 std::vector<int> GetOutputShape(int i) { return GetTensorShape(outputs_[i]); } in GetOutputShape()
76 std::vector<int> outputs_; member in tflite::__anon363ae4fb0111::SplitOpModel
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dapi.cc465 return LinkTensors(outputs, factory, &outputs_); in Initialize()
473 return GetExternalDefinitions(outputs_); in outputs()
485 if (index < 0 || index >= outputs_.size()) { in GetOutputObject()
488 *object = outputs_[index]->GetExternalObject(); in GetOutputObject()
500 if (index < 0 || index >= outputs_.size()) { in SetOutputObject()
503 return outputs_[index]->SetExternalObject(object); in SetOutputObject()
515 if (index > outputs_.size()) { in CopyToExternalOutput()
519 return outputs_[index]->CopyToExternalObject(); in CopyToExternalOutput()
534 for (int i = 0; i < outputs_.size(); i++) { in Run()
581 std::vector<std::unique_ptr<TensorTie>> outputs_; member in tflite::gpu::cl::__anon117e98fd0111::InferenceRunnerImpl
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_case_op.cc67 std::vector<NodeOut> outputs_; member in tensorflow::__anon74dd97450111::CaseBuilder
202 outputs_.resize(merges.size()); in AddOutputs()
214 outputs_[i] = NodeOut(merges[i], 0); in AddOutputs()
260 NodeBuilder builder = keep_node_fetchable_ && !outputs_.empty() in BuildLoweredCaseOutput()
261 ? NodeBuilder(name_, "IdentityN").Input(outputs_) in BuildLoweredCaseOutput()
Dlower_if_op.cc74 std::vector<NodeOut> outputs_; member in tensorflow::__anon5deaa8580111::CondBuilder
237 outputs_.resize(merges.size()); in AddOutputs()
245 outputs_[i] = NodeOut(merges[i], 0); in AddOutputs()
289 NodeBuilder builder = keep_node_fetchable_ && !outputs_.empty() in BuildLoweredIfOutput()
290 ? NodeBuilder(name_, "IdentityN").Input(outputs_) in BuildLoweredIfOutput()
/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
Dcluster.cc44 outputs_.push_back(node.input(i)); in FindClusterInputsAndOutputs()
49 return (!inputs_.empty()) && (!outputs_.empty()); in FindClusterInputsAndOutputs()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dtest_util.h57 return outputs_[index].data; in GetOutput()
63 std::vector<TensorFloat32> outputs_; variable
/external/tensorflow/tensorflow/lite/
Darena_planner_test.cc45 : inputs_(inputs), outputs_(outputs), temporaries_(temporaries) {} in TestOp()
48 const std::vector<int>& outputs() const { return outputs_; } in outputs()
53 std::vector<int> outputs_; member in tflite::__anond3fd21620111::TestOp
64 : inputs_(inputs), outputs_(outputs) { in TestGraph()
115 const std::vector<int>& outputs() { return outputs_; } in outputs()
126 std::swap(outputs_, other->outputs_); in Swap()
134 std::vector<int> outputs_; member in tflite::__anond3fd21620111::TestGraph
/external/tensorflow/tensorflow/lite/tools/evaluation/stages/
Dtflite_inference_stage.h74 const std::vector<void*>* GetOutputs() const { return &outputs_; } in GetOutputs()
87 std::vector<void*> outputs_; variable
Dtflite_inference_stage.cc46 outputs_.clear(); in UpdateModelInfo()
47 outputs_.reserve(interpreter_->outputs().size()); in UpdateModelInfo()
50 outputs_.push_back(tensor->data.raw); in UpdateModelInfo()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc330 : params_(params), outputs_(num_outputs) { in OpKernelContext()
348 for (TensorValue& value : outputs_) { in ~OpKernelContext()
476 outputs_[output_index] = TensorValue(new_tensor.release()); in forward_input_to_output_with_shape()
477 *output = outputs_[output_index].tensor; in forward_input_to_output_with_shape()
739 " num_outputs=", outputs_.size(), in allocate_output()
769 outputs_[index] = TensorValue(output_tensor.release()); in allocate_output()
770 *output = outputs_[index].tensor; in allocate_output()
942 outputs_[index] = TensorValue(new_tensor.release()); in maybe_set_output_by_allocate_and_copy()
968 CHECK_LT(index, outputs_.size()); in set_output()
971 CHECK_EQ(outputs_[index].tensor, nullptr); in set_output()
[all …]
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib.h91 outputs_(subgraph->outputs()), in SubgraphWriter()
126 outputs_(subgraph->outputs()), in SubgraphWriter()
189 std::vector<int> outputs_; variable
/external/tensorflow/tensorflow/lite/delegates/gpu/
Dgl_delegate.cc212 outputs_.clear(); in Prepare()
213 outputs_.reserve(delegate_params->output_tensors->size); in Prepare()
224 outputs_.push_back(output->id); in Prepare()
312 for (ValueId id : outputs_) { in Invoke()
364 std::vector<ValueId> outputs_; member in tflite::gpu::gl::__anon640f09d20111::Delegate
/external/tensorflow/tensorflow/java/src/gen/cc/
Dop_specs.h148 const std::vector<ArgumentSpec>& outputs() const { return outputs_; } in outputs()
171 std::vector<ArgumentSpec> outputs_; variable
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Dapi2.cc507 outputs_ = LinkTensors(graph_.outputs()); in Initialize()
516 return GetExternalDefinitions(outputs_); in outputs()
541 if (index < 0 || index >= outputs_.size()) { in SetOutputObjectDef()
544 auto def = outputs_[index]; in SetOutputObjectDef()
550 outputs_[index] = def; in SetOutputObjectDef()
582 RETURN_IF_ERROR(runner_impl->Initialize(inputs_, outputs_, &tie_factory_)); in Build()
655 std::vector<TensorTieDef> outputs_; member in tflite::gpu::gl::__anon517d4d930111::InferenceBuilderImpl

12