| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 116 outputs_.push_back(index); in OpOutputs()
118 vector_.resize(outputs_.size()); in OpOutputs()
126 for (int i : outputs_) { in InitializeGraphOutputs()
149 int Size() const { return outputs_.size(); } in Size()
151 int TfLiteIndex(int i) const { return outputs_[i]; } in TfLiteIndex()
169 std::vector<int> outputs_; member in tflite::flex::kernel::OpOutputs
179 : inputs_(inputs), outputs_(outputs) {} in OpNode()
195 const OpOutputs& outputs() const { return outputs_; } in outputs()
196 OpOutputs* mutable_outputs() { return &outputs_; } in mutable_outputs()
199 int NumOutputs() const { return outputs_.Size(); } in NumOutputs()
[all …]
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_beam_search_decoder_test.cc | 51 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::__anon785e371d0111::CTCBeamSearchDecoderOpModel
|
| /external/tensorflow/tensorflow/c/ |
| D | while_loop_test.cc | 58 outputs_.resize(ninputs, {nullptr, -1}); in Init()
62 TF_FinishWhile(params_.get(), s_, &outputs_[0]); in ExpectOK()
67 TF_FinishWhile(params_.get(), s_, &outputs_[0]); in ExpectError()
76 Run(outputs_, input_values); in Run()
129 std::vector<TF_Output> outputs_; // The final outputs of the while loop member in __anona1d00fec0111::CApiWhileLoopTest
178 EXPECT_TRUE(outputs_[0].oper != nullptr); in TEST_F()
179 EXPECT_GE(outputs_[0].index, 0); in TEST_F()
180 EXPECT_TRUE(outputs_[1].oper != nullptr); in TEST_F()
181 EXPECT_GE(outputs_[1].index, 0); in TEST_F()
434 TF_AddGradients(graph_, outputs_.data(), outputs_.size(), inputs_.data(), 1, in TEST_F()
|
| D | c_test_util.cc | 481 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/ |
| D | while_loop_test.cc | 43 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::__anon371ba7300111::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/ |
| D | while_gradients_test.cc | 45 &outputs_)); in CreateLoop()
50 AddSymbolicGradients(scope_, outputs_, inputs_, &grad_outputs_)); in CreateBackprop()
87 std::vector<Output> outputs_; member in tensorflow::__anonb1a79a530111::WhileGradientsTest
182 outputs_[0] = ops::Neg(scope_, outputs_[0]); in TEST_F()
213 AddSymbolicGradients(cpu1_scope, outputs_, inputs_, &grad_outputs_)); in TEST_F()
|
| D | gradients.cc | 103 const std::vector<Output>& outputs_; member in tensorflow::__anon810028dc0111::SymbolicGradientBuilder
146 outputs_(outputs), in SymbolicGradientBuilder()
170 for (const Output& out : outputs_) { in GetReachableNodes()
250 if (outputs_.size() != grad_inputs_.size()) { in Initialize()
267 output_nodes.reserve(outputs_.size()); in Initialize()
268 for (size_t i = 0; i < outputs_.size(); ++i) { in Initialize()
269 output_nodes.insert(outputs_[i].node()->id()); in Initialize()
322 for (const Output& output : outputs_) { in Initialize()
336 TF_RETURN_IF_ERROR(BackpropAlongEdge(grad_inputs_[i], outputs_[i])); in Initialize()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_if_op.cc | 70 std::vector<NodeOut> outputs_; member in tensorflow::__anoneb751e910111::CondBuilder
198 outputs_.resize(merges.size()); in AddOutputs()
206 outputs_[i] = NodeOut(merges[i], 0); in AddOutputs()
274 NodeBuilder ib(name_, outputs_.empty() ? "NoOp" : "IdentityN"); in BuildLoweredIfOutput()
275 if (!outputs_.empty()) ib.Input(outputs_); in BuildLoweredIfOutput()
284 if (outputs_.empty()) return Status::OK(); in InlineCallNodes()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | split_test.cc | 39 outputs_.push_back(AddOutput(input.type)); in SplitOpModel()
58 return ExtractVector<T>(outputs_[i]); in GetOutput()
60 std::vector<int> GetOutputShape(int i) { return GetTensorShape(outputs_[i]); } in GetOutputShape()
65 std::vector<int> outputs_; member in tflite::__anon0e1332340111::SplitOpModel
|
| D | split_v_test.cc | 41 outputs_.push_back(AddOutput(input.type)); in SplitVOpModel()
64 return ExtractVector<T>(outputs_[i]); in GetOutput()
66 std::vector<int> GetOutputShape(int i) { return GetTensorShape(outputs_[i]); } in GetOutputShape()
72 std::vector<int> outputs_; member in tflite::__anon6dfc68290111::SplitVOpModel
|
| D | test_util.cc | 68 outputs_.push_back(id); in AddOutput()
78 builder_.CreateVector<int32_t>(outputs_), builtin_options_type, in SetBuiltinOp()
91 builder_.CreateVector<int32_t>(outputs_), BuiltinOptions_NONE, 0, in SetCustomOp()
102 auto outputs = builder_.CreateVector<int32_t>(outputs_); in BuildInterpreter()
|
| D | unpack_test.cc | 37 outputs_.push_back(AddOutput(input.type)); in UnpackOpModel()
50 for (const int output : outputs_) { in GetOutputDatas()
59 for (const int output : outputs_) { in GetOutputShapes()
67 std::vector<int> outputs_; member in tflite::__anon72f3964a0111::UnpackOpModel
|
| /external/tensorflow/tensorflow/lite/ |
| D | arena_planner_test.cc | 34 : inputs_(inputs), outputs_(outputs), temporaries_(temporaries) {} in TestOp()
37 const std::vector<int>& outputs() const { return outputs_; } in outputs()
42 std::vector<int> outputs_; member in tflite::__anonabd56e9b0111::TestOp
53 : inputs_(inputs), outputs_(outputs) { in TestGraph()
104 const std::vector<int>& outputs() { return outputs_; } in outputs()
115 std::swap(outputs_, other->outputs_); in Swap()
123 std::vector<int> outputs_; member in tflite::__anonabd56e9b0111::TestGraph
|
| D | graph_info_test.cc | 47 const std::vector<int>& outputs() const override { return outputs_; } in outputs()
63 outputs_ = outputs; in SetInputsAndOutputs()
70 std::vector<int> outputs_; member in tflite::__anon6e38e6c20111::SimpleTestGraph
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.cc | 86 outputs_[NodeName(input)].insert(nodes_[node_name]); in NodeMap()
93 outputs_.erase(NodeName(name)); in RemoveNode()
111 auto it = outputs_.find(node_name); in GetOutputs()
112 if (it == outputs_.end()) { in GetOutputs()
128 outputs_[node_name].insert(output_node); in AddOutput()
132 outputs_[node_name].erase(nodes_[NodeName(output_name)]); in RemoveOutput()
149 outputs_.erase(node_name); in RemoveOutputs()
155 std::set<NodeDef*>& outputs = outputs_[node_name]; in UpdateOutput()
|
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | cluster.cc | 44 outputs_.push_back(node.input(i)); in FindClusterInputsAndOutputs()
49 return (!inputs_.empty()) && (!outputs_.empty()); in FindClusterInputsAndOutputs()
|
| D | cluster.h | 69 std::vector<string> outputs_; variable
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| D | beam_search_decoder_test.py | 302 outputs_, next_state_, state_, log_probs_ = sess.run(
305 self.assertAllEqual(outputs_.predicted_ids, [[3, 3, 2], [2, 2, 1]])
306 self.assertAllEqual(outputs_.parent_ids, [[1, 0, 0], [2, 1, 0]])
359 outputs_, next_state_, state_, log_probs_ = sess.run(
362 self.assertAllEqual(outputs_.parent_ids, [[1, 0, 0], [1, 2, 0]])
363 self.assertAllEqual(outputs_.predicted_ids, [[0, 3, 2], [2, 0, 1]])
460 outputs_, next_state_, _, _ = sess.run(
463 self.assertEqual(outputs_.predicted_ids[0, 0], 3)
464 self.assertEqual(outputs_.predicted_ids[0, 1], 2)
465 self.assertEqual(outputs_.predicted_ids[1, 0], 1)
|
| /external/tensorflow/tensorflow/java/src/gen/cc/ |
| D | op_specs.h | 148 const std::vector<ArgumentSpec>& outputs() const { return outputs_; } in outputs()
171 std::vector<ArgumentSpec> outputs_; variable
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 283 outputs_(num_outputs), in OpKernelContext()
302 for (TensorValue& value : outputs_) { in ~OpKernelContext()
460 outputs_[output_index] = TensorValue(new_tensor.release()); in forward_input_to_output_with_shape()
461 *output = outputs_[output_index].tensor; in forward_input_to_output_with_shape()
729 DCHECK_LT(index, outputs_.size()); in allocate_output()
736 outputs_[index] = TensorValue(output_tensor.release()); in allocate_output()
737 *output = outputs_[index].tensor; in allocate_output()
798 DCHECK_LT(index, outputs_.size()); in set_output()
802 outputs_[index] = TensorValue(new Tensor(tensor)); in set_output()
824 DCHECK_LT(index, outputs_.size()); in set_output_ref()
[all …]
|
| D | shape_inference.h | 326 ShapeHandle output(int64 idx) const { return outputs_.at(idx); } in output()
327 void set_output(int idx, ShapeHandle shape) { outputs_.at(idx) = shape; } in set_output()
331 int num_outputs() const { return outputs_.size(); } in num_outputs()
332 ShapeHandle output(int idx) const { return outputs_.at(idx); } in output()
753 std::vector<ShapeHandle> outputs_; variable
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.h | 150 std::vector<int>& outputs() { return outputs_; } in outputs()
153 const std::vector<int>& outputs() const { return outputs_; } in outputs()
485 std::vector<int> outputs_; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils_test.cc | 96 graph_def_with_shapetype, inputs_, outputs_, in FuseByInOut()
104 graph_def_, inputs_, outputs_, "remote_fused_graph_node_names", in FuseByNodes()
111 graph_def_, inputs_, outputs_, "remote_fused_graph_node_names", in FuseByOpTypes()
118 graph_def_, inputs_, outputs_, REMOTE_FUSED_EXECUTOR_NAME0, in FuseByExecutor0()
124 graph_def_, inputs_, outputs_, REMOTE_FUSED_EXECUTOR_NAME1, in FuseByExecutor1()
135 inputs_, outputs_, subgraph_node_names_, subgraph_input_names_, in PlaceRemoteGraphArguments()
167 const std::vector<string> outputs_{"K"}; member in tensorflow::__anon29f32cd80111::FuseRemoteGraphMultipleAddOpsTest
|
| D | remote_fused_graph_rewriter_transform_test.cc | 108 context.output_names = outputs_; in FuseInternal()
204 const std::vector<string> outputs_{"K"}; member in tensorflow::graph_transforms::__anon5181c9840111::FuseRemoteGraphMultipleAddOpsRewriterTest
|
| /external/tensorflow/tensorflow/core/util/ |
| D | stat_summarizer.cc | 180 outputs_.emplace(name, std::vector<TensorDescription>()); in ProcessStepStats()
231 const auto detail_outputs = outputs_.at(entry.second->first); in PrintOutputs()
|