Home
last modified time | relevance | path

Searched refs:inputs_ (Results 1 – 25 of 61) sorted by relevance

123

/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/
Dvectorizer.h38 : inputs_(std::move(inputs)) {} in VectorizerInput()
48 if (!inputs_[index].stacked) { in stacked()
52 *result = {inputs_[index].node, inputs_[index].output_index}; in stacked()
64 if (inputs_[index].stacked) { in unstacked()
68 *result = {inputs_[index].node, inputs_[index].output_index}; in unstacked()
76 return inputs_.at(index); in at()
81 return inputs_.begin(); in begin()
85 return inputs_.end(); in end()
89 size_t size() const { return inputs_.size(); } in size()
92 std::vector<WrappedTensor> inputs_;
/external/tensorflow/tensorflow/cc/framework/
Dwhile_gradients_test.cc36 inputs_.push_back(ops::Placeholder(scope_, dtype)); in Init()
43 if (inputs == nullptr) inputs = &inputs_; in CreateLoop()
50 AddSymbolicGradients(scope_, outputs_, inputs_, &grad_outputs_)); in CreateBackprop()
51 ASSERT_EQ(grad_outputs_.size(), inputs_.size()); in CreateBackprop()
66 DCHECK_EQ(input_values.size(), inputs_.size()); in Run()
68 for (int i = 0; i < inputs_.size(); ++i) { in Run()
69 feeds.emplace(inputs_[i], input_values[i]); in Run()
86 std::vector<Output> inputs_; member in tensorflow::__anon27dc20fc0111::WhileGradientsTest
163 std::vector<Output> loop_inputs = {ops::Multiply(scope_, inputs_[0], 2.0), in TEST_F()
164 ops::Multiply(scope_, inputs_[1], 2.0)}; in TEST_F()
[all …]
Dops.h60 Inputs inputs_; variable
263 inputs_.push_back(x);
269 : inputs_(inputs.begin(), inputs.end()) {}
273 : inputs_(inputs.begin(), inputs.end()) {}
278 inputs_.push_back(x);
282 typename std::vector<Input>::iterator begin() { return inputs_.begin(); }
283 typename std::vector<Input>::iterator end() { return inputs_.end(); }
285 return inputs_.begin();
288 return inputs_.end();
292 std::vector<Input> inputs_;
Dops.cc21 Operation::Operation(Node* n) : inputs_(GetInputs(n)), node_(n) {} in Operation()
29 if (inputs_[i].first == nullptr && inputs_[i].second == -1) { in input()
37 return Output(inputs_[i].first, inputs_[i].second); in input()
/external/tensorflow/tensorflow/core/kernels/
Dops_testutil.h140 CHECK_GT(input_types_.size(), inputs_.size()) in AddResourceInput()
156 inputs_.push_back({nullptr, input}); in AddResourceInput()
170 params_.get()->inputs = &inputs_; in RunOpKernel()
195 CHECK_LT(input_index, inputs_.size()); in mutable_input()
196 return inputs_[input_index]; in mutable_input()
209 CHECK_GT(input_types_.size(), inputs_.size()) in AddInput()
211 bool is_ref = IsRefType(input_types_[inputs_.size()]); in AddInput()
215 CHECK_EQ(RemoveRefType(input_types_[inputs_.size()]), dtype); in AddInput()
216 inputs_.push_back({&lock_for_refs_, input}); in AddInput()
218 CHECK_EQ(input_types_[inputs_.size()], dtype); in AddInput()
[all …]
Dquantize_and_dequantize_op_test.cc62 EXPECT_EQ(inputs_[1]->scalar<float>()(), 0.0); in TEST_F()
63 EXPECT_EQ(inputs_[2]->scalar<float>()(), 0.0); in TEST_F()
88 EXPECT_EQ(inputs_[1]->scalar<float>()(), 0.0); in TEST_F()
89 EXPECT_EQ(inputs_[2]->scalar<float>()(), 0.0); in TEST_F()
119 EXPECT_EQ(inputs_[1]->scalar<float>()(), 0.0); in TEST_F()
120 EXPECT_EQ(inputs_[2]->scalar<float>()(), 0.0); in TEST_F()
152 EXPECT_EQ(inputs_[1]->scalar<float>()(), 0.0); in TEST_F()
153 EXPECT_EQ(inputs_[2]->scalar<float>()(), 0.0); in TEST_F()
183 EXPECT_EQ(inputs_[1]->scalar<float>()(), 0.0); in TEST_F()
184 EXPECT_EQ(inputs_[2]->scalar<float>()(), 0.0); in TEST_F()
[all …]
Dresize_area_op_test.cc38 inputs_.clear(); in SetRandomImageInput()
41 bool is_ref = IsRefType(input_types_[inputs_.size()]); in SetRandomImageInput()
47 CHECK_EQ(RemoveRefType(input_types_[inputs_.size()]), in SetRandomImageInput()
49 inputs_.push_back({&lock_for_refs_, input}); in SetRandomImageInput()
51 CHECK_EQ(input_types_[inputs_.size()], DataTypeToEnum<float>::v()); in SetRandomImageInput()
52 inputs_.push_back({nullptr, input}); in SetRandomImageInput()
/external/tensorflow/tensorflow/core/kernels/data/
Dzip_dataset_op.cc45 : DatasetBase(DatasetContext(ctx)), inputs_(inputs) { in Dataset()
46 for (const auto& input : inputs_) { in Dataset()
58 for (const auto& input : inputs_) { in ~Dataset()
81 for (const auto& input : inputs_) { in Cardinality()
99 input_graph_nodes.reserve(inputs_.size()); in AsGraphDefInternal()
100 for (const auto& input : inputs_) { in AsGraphDefInternal()
118 input_impls_.resize(dataset()->inputs_.size()); in Initialize()
120 TF_RETURN_IF_ERROR(dataset()->inputs_[i]->MakeIterator( in Initialize()
180 DCHECK_EQ(input_impls_.size(), dataset()->inputs_.size()); in RestoreInternal()
192 const std::vector<DatasetBase*> inputs_; member in tensorflow::data::__anon8a8355ec0111::ZipDatasetOp::Dataset
/external/v8/src/compiler/
Dnode.h63 : inputs_.outline_->count_; in InputCount()
172 Node* inputs_[1]; member
191 ? reinterpret_cast<Node*>(start)->inputs_.inline_ in input_ptr()
192 : reinterpret_cast<OutOfLineInputs*>(start)->inputs_; in input_ptr()
242 return has_inline_inputs() ? &(inputs_.inline_[input_index]) in GetInputPtrConst()
243 : &inputs_.outline_->inputs_[input_index]; in GetInputPtrConst()
246 return has_inline_inputs() ? &(inputs_.inline_[input_index]) in GetInputPtr()
247 : &inputs_.outline_->inputs_[input_index]; in GetInputPtr()
251 : reinterpret_cast<Use*>(inputs_.outline_); in GetUsePtr()
293 } inputs_; variable
[all …]
Dnode.cc28 Node** new_input_ptr = inputs_; in ExtractFrom()
78 node->inputs_.outline_ = outline; in New()
83 input_ptr = outline->inputs_; in New()
100 input_ptr = node->inputs_.inline_; in New()
122 ? node->inputs_.inline_ in Clone()
123 : node->inputs_.outline_->inputs_; in Clone()
161 inputs_.outline_ = outline; in AppendInput()
164 outline = inputs_.outline_; in AppendInput()
170 inputs_.outline_ = outline; in AppendInput()
250 inputs_.outline_->count_ = new_input_count; in TrimInputCount()
/external/tensorflow/tensorflow/core/framework/
Dmodel.cc78 if (inputs_.size() <= 1) { in OutputTimeLocked()
81 int64 delta = NanosPerElementLocked() * (inputs_.size() - 1); in OutputTimeLocked()
87 inputs_.front()->OutputTime(input_times)) / in OutputTimeLocked()
88 static_cast<double>(inputs_.size() - 1); in OutputTimeLocked()
93 if (inputs_.size() <= 1) { in ProcessingTimeLocked()
98 inputs_.front()->ProcessingTime()) / in ProcessingTimeLocked()
99 static_cast<double>(inputs_.size() - 1); in ProcessingTimeLocked()
130 if (inputs_.size() <= 1) { in OutputTimeLocked()
135 static_cast<double>(inputs_.size() - 1); in OutputTimeLocked()
139 double parallelism = inputs_.size() - 1; // default to cycle length in OutputTimeLocked()
[all …]
Dmodel.h130 inputs_.push_back(node); in add_input()
160 return inputs_; in inputs()
210 inputs_.remove(input); in remove_input()
223 for (auto& input : inputs_) { in CollectTunableParameters()
257 for (auto& input : inputs_) { in Snapshot()
281 for (auto& input : inputs_) { in OutputTimeForInputs()
296 for (auto& input : inputs_) { in ProcessingTimeForInputs()
314 std::list<std::shared_ptr<Node>> inputs_ GUARDED_BY(mu_);
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dchoose_fastest_dataset_op.cc123 inputs_(std::move(inputs)), in Dataset()
128 for (auto input : inputs_) { in Dataset()
134 for (auto input : inputs_) { in ~Dataset()
165 input_nodes.reserve(inputs_.size()); in AsGraphDefInternal()
166 for (const auto& input : inputs_) { in AsGraphDefInternal()
184 histograms_(dataset()->inputs_.size()) {} in ChooseFastestIterator()
188 input_impls_.resize(dataset()->inputs_.size()); in Initialize()
189 for (size_t i = 0, num_inputs = dataset()->inputs_.size(); in Initialize()
191 TF_RETURN_IF_ERROR(dataset()->inputs_[i]->MakeIterator( in Initialize()
260 TF_RETURN_IF_ERROR(dataset()->inputs_[fastest_index_]->MakeIterator( in RestoreInternal()
[all …]
/external/tensorflow/tensorflow/lite/testing/kernel_test/
Dinput_generator.cc112 inputs_.push_back(input); in ReadInputsFromFile()
131 for (const auto& input : inputs_) { in WriteInputsToFile()
150 inputs_.push_back(Join(data.data(), data.size(), ",")); in GenerateInput()
157 inputs_.push_back(Join(data.data(), data.size(), ",")); in GenerateInput()
162 inputs_.push_back(JoinDefault(data.data(), data.size(), ",")); in GenerateInput()
176 inputs_.push_back(Join(data.data(), data.size(), ",")); in GenerateInput()
183 inputs_.push_back(Join(data.data(), data.size(), ",")); in GenerateInput()
188 inputs_.push_back(JoinDefault(data.data(), data.size(), ",")); in GenerateInput()
205 std::vector<string> InputGenerator::GetInputs() { return inputs_; } in GetInputs()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc68 inputs_.push_back(index); in OpInputs()
70 forwardable_.resize(inputs_.size()); in OpInputs()
74 int Size() const { return inputs_.size(); } in Size()
76 int TfLiteIndex(int i) const { return inputs_[i]; } in TfLiteIndex()
83 for (int i : inputs_) { in InitializeTensorSources()
100 std::vector<int> inputs_; member in tflite::flex::kernel::OpInputs
179 : inputs_(inputs), outputs_(outputs) {} in OpNode()
192 const OpInputs& inputs() const { return inputs_; } in inputs()
193 OpInputs* mutable_inputs() { return &inputs_; } in mutable_inputs()
198 int NumInputs() const { return inputs_.Size(); } in NumInputs()
[all …]
/external/tensorflow/tensorflow/lite/profiling/
Dprofile_summarizer_test.cc79 PopulateTensor(inputs_[0], {x}); in SetInputs()
80 PopulateTensor(inputs_[1], {y}); in SetInputs()
85 int inputs_[2]; member in tflite::profiling::__anon9090bee60111::SimpleOpModel
91 inputs_[0] = AddInput({TensorType_INT32, {1}}); in Init()
92 inputs_[1] = AddInput({TensorType_INT32, {1}}); in Init()
95 BuildInterpreter({GetShape(inputs_[0]), GetShape(inputs_[1])}); in Init()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_operation.cc30 inputs_.push_back(h); in AddInput()
31 attrs_.NumInputs(static_cast<int>(inputs_.size())); in AddInput()
35 inputs_.push_back(h); in ConsumeInput()
36 attrs_.NumInputs(static_cast<int>(inputs_.size())); in ConsumeInput()
46 for (const auto& input : inputs_) { in DebugString()
Dexecute_node.h42 inputs_(inputs), in ExecuteNode()
48 for (auto handle : inputs_) { in ExecuteNode()
57 for (auto handle : inputs_) { in ~ExecuteNode()
67 ctx_, inputs_, kernel_, maybe_stats_.get(), maybe_step_stats_, in Run()
81 tensorflow::gtl::InlinedVector<TensorHandle*, 4> inputs_; variable
Deager_operation.h35 for (tensorflow::TensorHandle* h : inputs_) { in ~EagerOperation()
49 return inputs_; in Inputs()
53 return &inputs_; in MutableInputs()
74 tensorflow::gtl::InlinedVector<tensorflow::TensorHandle*, 4> inputs_; variable
/external/tensorflow/tensorflow/c/
Dwhile_loop_test.cc37 DCHECK(inputs_.empty()); in Init()
44 inputs_.push_back({placeholder, 0}); in Init()
50 TF_NewWhile(graph_, &inputs_[0], inputs_.size(), s_))); in Init()
81 DCHECK_EQ(inputs_.size(), input_values.size()); in Run()
82 std::vector<std::pair<TF_Operation*, TF_Tensor*>> inputs(inputs_.size()); in Run()
85 inputs[i] = {inputs_[i].oper, Int32Tensor(v)}; in Run()
128 std::vector<TF_Output> inputs_; // The inputs to the while loop member in __anonfc0664950111::CApiWhileLoopTest
316 params_->cond_output = inputs_[0]; in TEST_F()
357 params_->body_outputs[0] = inputs_[0]; in TEST_F()
389 params_->body_outputs[0] = inputs_[0]; in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/graph/
Dnode_builder.cc52 inputs_.emplace_back(src_node, src_index); in Input()
64 inputs_.emplace_back(src.node, src.index); in Input()
78 inputs_.emplace_back(node_out.node, node_out.index); in Input()
133 for (size_t i = 0; i < inputs_.size(); ++i) { in Finalize()
134 if (inputs_[i].node != nullptr) { // Skip back edges. in Finalize()
135 graph->AddEdge(inputs_[i].node, inputs_[i].index, node, i); in Finalize()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_execute_node.h40 inputs_(inputs), in RemoteExecuteNode()
42 for (auto* handle : inputs_) { in RemoteExecuteNode()
55 for (auto* handle : inputs_) { in ~RemoteExecuteNode()
84 gtl::InlinedVector<TensorHandle*, 4> inputs_; variable
/external/tensorflow/tensorflow/lite/kernels/
Dadd_n_test.cc36 inputs_.push_back(AddInput(inputs[i])); in BaseAddNOpModel()
37 input_shapes.push_back(GetShape(inputs_[i])); in BaseAddNOpModel()
46 int input(int i) { return inputs_[i]; } in input()
49 std::vector<int> inputs_; member in tflite::__anonb7999aa70111::BaseAddNOpModel
/external/tensorflow/tensorflow/cc/ops/
Dwhile_loop_test.cc34 inputs_.push_back(ops::Placeholder(scope_, dtype)); in Init()
43 ops::BuildWhileLoop(scope_, inputs_, cond, body, kFrameName, &outputs_); in CreateLoop()
53 DCHECK_EQ(input_values.size(), inputs_.size()); in Run()
55 for (int i = 0; i < inputs_.size(); ++i) { in Run()
56 feeds.emplace(inputs_[i], input_values[i]); in Run()
71 std::vector<Output> inputs_; member in tensorflow::__anonde9e4d790111::WhileLoopTest
/external/libtextclassifier/lang_id/common/fel/
Dtask-context.cc27 auto it = inputs_.find(name); in GetInputPath()
28 if (it != inputs_.end()) { in GetInputPath()
35 inputs_[name] = path; in SetInputPath()

123