| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.cc | 44 outputs_[NodeName(input)].insert(nodes_[node_name]); in NodeMap()
51 outputs_.erase(NodeName(name)); in RemoveNode()
69 auto it = outputs_.find(node_name); in GetOutputs()
70 if (it == outputs_.end()) { in GetOutputs()
86 outputs_[node_name].insert(output_node); in AddOutput()
90 outputs_[node_name].erase(nodes_[NodeName(output_name)]); in RemoveOutput()
107 outputs_.erase(node_name); in RemoveOutputs()
113 std::set<NodeDef*>& outputs = outputs_[node_name]; in UpdateOutput()
342 outputs_.clear(); in Initialize()
343 outputs_.resize(num_nodes); in Initialize()
[all …]
|
| D | utils.h | 59 std::unordered_map<string, std::set<NodeDef*>> outputs_; variable
191 return outputs_[node_idx]; in outputs()
200 std::vector<gtl::InlinedVector<int, 2>> outputs_; variable
|
| /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 __anon74c604890111::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 | 433 outputs_.clear(); in SetOutputs()
435 outputs_.emplace_back(TF_Output{o, 0}); in SetOutputs()
437 output_values_.resize(outputs_.size()); in SetOutputs()
442 outputs_ = outputs; in SetOutputs()
443 output_values_.resize(outputs_.size()); in SetOutputs()
459 output_values_.resize(outputs_.size(), nullptr); in Run()
465 const TF_Output* outputs_ptr = outputs_.empty() ? nullptr : &outputs_[0]; in Run()
472 outputs_ptr, output_values_ptr, outputs_.size(), targets_ptr, in Run()
|
| D | c_test_util.h | 144 std::vector<TF_Output> outputs_; variable
|
| /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::__anonc42405ed0111::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::__anon89a528f00111::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::__anon0e0887990111::SymbolicGradientBuilder
146 outputs_(outputs), in SymbolicGradientBuilder()
171 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/contrib/lite/kernels/ |
| D | split_test.cc | 39 outputs_.push_back(AddOutput(input.type)); in SplitOpModel()
56 return ExtractVector<float>(outputs_[i]); in GetOutput()
58 std::vector<int> GetOutputShape(int i) { return GetTensorShape(outputs_[i]); } in GetOutputShape()
63 std::vector<int> outputs_; member in tflite::__anon02dcc3910111::SplitOpModel
|
| D | test_util.cc | 127 outputs_.push_back(id); in AddOutput()
137 builder_.CreateVector<int32_t>(outputs_), builtin_options_type, in SetBuiltinOp()
150 builder_.CreateVector<int32_t>(outputs_), BuiltinOptions_NONE, 0, in SetCustomOp()
161 auto outputs = builder_.CreateVector<int32_t>(outputs_); in BuildInterpreter()
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| D | beam_search_decoder_test.py | 152 outputs_, next_state_, state_, log_probs_ = sess.run(
155 self.assertAllEqual(outputs_.predicted_ids, [[3, 3, 2], [2, 2, 1]])
156 self.assertAllEqual(outputs_.parent_ids, [[1, 0, 0], [2, 1, 0]])
207 outputs_, next_state_, state_, log_probs_ = sess.run(
210 self.assertAllEqual(outputs_.parent_ids, [[1, 0, 0], [1, 2, 0]])
211 self.assertAllEqual(outputs_.predicted_ids, [[0, 3, 2], [2, 0, 1]])
305 outputs_, next_state_, _, _ = sess.run(
308 self.assertEqual(outputs_.predicted_ids[0, 0], 3)
309 self.assertEqual(outputs_.predicted_ids[0, 1], 2)
310 self.assertEqual(outputs_.predicted_ids[1, 0], 1)
|
| /external/tensorflow/tensorflow/contrib/lite/ |
| D | arena_planner_test.cc | 32 : inputs_(inputs), outputs_(outputs), temporaries_(temporaries) {} in TestOp()
35 const std::vector<int>& outputs() const { return outputs_; } in outputs()
40 std::vector<int> outputs_; member in tflite::__anona09efff80111::TestOp
51 : inputs_(inputs), outputs_(outputs) { in TestGraph()
102 const std::vector<int>& outputs() { return outputs_; } in outputs()
108 std::vector<int> outputs_; member in tflite::__anona09efff80111::TestGraph
|
| D | interpreter.h | 161 const std::vector<int>& outputs() const { return outputs_; } in outputs()
166 return context_.tensors[outputs_[index]].name; in GetOutputName()
223 return typed_tensor<T>(outputs_[index]); in typed_output_tensor()
398 std::vector<int> outputs_; variable
|
| D | graph_info_test.cc | 47 const std::vector<int>& outputs() const override { return outputs_; } in outputs()
62 outputs_ = outputs; in SetInputsAndOutputs()
69 std::vector<int> outputs_; member in tflite::__anonab6bad3f0111::SimpleTestGraph
|
| /external/tensorflow/tensorflow/contrib/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/core/framework/ |
| D | op_kernel.cc | 262 outputs_(num_outputs), in OpKernelContext()
276 for (TensorValue& value : outputs_) { in ~OpKernelContext()
422 outputs_[output_index] = TensorValue(new_tensor.release()); in forward_input_to_output_with_shape()
423 *output = outputs_[output_index].tensor; in forward_input_to_output_with_shape()
653 DCHECK_LT(index, outputs_.size()); in allocate_output()
660 outputs_[index] = TensorValue(output_tensor); in allocate_output()
661 *output = outputs_[index].tensor; in allocate_output()
722 DCHECK_LT(index, outputs_.size()); in set_output()
726 outputs_[index] = TensorValue(new Tensor(tensor)); in set_output()
748 DCHECK_LT(index, outputs_.size()); in set_output_ref()
[all …]
|
| D | shape_inference.h | 320 void set_output(int idx, ShapeHandle shape) { outputs_[idx] = shape; } in set_output()
324 int num_outputs() const { return outputs_.size(); } in num_outputs()
325 ShapeHandle output(int idx) const { return outputs_[idx]; } in output()
717 std::vector<ShapeHandle> outputs_; variable
|
| D | op_def_builder.h | 158 std::vector<string> outputs_; variable
|
| D | op_kernel.h | 588 int num_outputs() const { return outputs_.size(); } in num_outputs()
1107 gtl::InlinedVector<TensorValue, 4> outputs_; variable
1378 return outputs_[index].tensor; in mutable_output()
1384 TensorValue value = outputs_[index]; in release_output()
1385 outputs_[index] = TensorValue(); in release_output()
|
| D | op_def_builder.cc | 539 outputs_.emplace_back(spec.data(), spec.size()); in Output()
609 for (StringPiece output : outputs_) { in Finalize()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils_test.cc | 95 graph_def_with_shapetype, inputs_, outputs_, in FuseByInOut()
103 graph_def_, inputs_, outputs_, "remote_fused_graph_node_names", in FuseByNodes()
110 graph_def_, inputs_, outputs_, "remote_fused_graph_node_names", in FuseByOpTypes()
117 graph_def_, inputs_, outputs_, REMOTE_FUSED_EXECUTOR_NAME0, in FuseByExecutor0()
123 graph_def_, inputs_, outputs_, REMOTE_FUSED_EXECUTOR_NAME1, in FuseByExecutor1()
134 inputs_, outputs_, subgraph_node_names_, subgraph_input_names_, in PlaceRemoteGraphArguments()
166 const std::vector<string> outputs_{"K"}; member in tensorflow::__anone61169150111::FuseRemoteGraphMultipleAddOpsTest
|
| D | remote_fused_graph_rewriter_transform_test.cc | 106 context.output_names = outputs_; in FuseInternal()
202 const std::vector<string> outputs_{"K"}; member in tensorflow::graph_transforms::__anond43033e10111::FuseRemoteGraphMultipleAddOpsRewriterTest
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| D | helper.py | 446 def maybe_concatenate_auxiliary_inputs(outputs_, indices=None): argument
449 return outputs_
458 outputs_, auxiliary_inputs)
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_constructor.cc | 289 std::vector<gtl::InlinedVector<int, 4>> outputs_; member in tensorflow::__anond2858b560111::GraphConstructor
470 outputs_.resize(num_nodes); in InitFromEdges()
513 outputs_[iter->second.gdef_index].push_back(n); in InitFromEdges()
921 UpdatePendingCountAndReady(outputs_, o, &pending_count_, &ready_); in Convert()
1020 UpdatePendingCountAndReady(outputs_, o, &pending_count_, &ready_); in Convert()
|