Home
last modified time | relevance | path

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

1234

/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::__anon97a6b1da0111::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()
Dgradients.cc105 const std::vector<Output>& inputs_; member in tensorflow::__anonc8091ec30111::SymbolicGradientBuilder
148 inputs_(inputs), in SymbolicGradientBuilder()
203 for (const Output& nout : inputs_) { in GetStopBackpropNodes()
256 for (const Output& input : inputs_) { in Initialize()
265 grad_outputs_->resize(inputs_.size()); in Initialize()
277 input_nodes_.reserve(inputs_.size()); in Initialize()
278 for (size_t i = 0; i < inputs_.size(); ++i) { in Initialize()
279 input_nodes_.insert({inputs_[i], i}); in Initialize()
289 for (const Output& nout : inputs_) { in Initialize()
540 for (const Output& nout : inputs_) { in AddGradients()
/external/tensorflow/tensorflow/core/kernels/data/
Dzip_dataset_op.cc39 : DatasetBase(DatasetContext(ctx)), inputs_(inputs) { in Dataset()
40 for (const auto& input : inputs_) { in Dataset()
52 for (const auto& input : inputs_) { in ~Dataset()
77 for (const auto& input : inputs_) { in Cardinality()
91 for (const auto& input : inputs_) { in InputDatasets()
98 for (const auto& input : inputs_) { in CheckExternalState()
109 input_graph_nodes.reserve(inputs_.size()); in AsGraphDefInternal()
110 for (const auto& input : inputs_) { in AsGraphDefInternal()
128 input_impls_.resize(dataset()->inputs_.size()); in Initialize()
130 TF_RETURN_IF_ERROR(dataset()->inputs_[i]->MakeIterator( in Initialize()
[all …]
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dchoose_fastest_dataset_op.cc124 inputs_(std::move(inputs)), in Dataset()
129 for (auto input : inputs_) { in Dataset()
135 for (auto input : inputs_) { in ~Dataset()
163 for (const auto& input : inputs_) { in InputDatasets()
170 for (const auto& input : inputs_) { in CheckExternalState()
181 input_nodes.reserve(inputs_.size()); in AsGraphDefInternal()
182 for (const auto& input : inputs_) { in AsGraphDefInternal()
200 histograms_(dataset()->inputs_.size()) {} in ChooseFastestIterator()
204 input_impls_.resize(dataset()->inputs_.size()); in Initialize()
205 for (size_t i = 0, num_inputs = dataset()->inputs_.size(); in Initialize()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute_node.h104 inputs_(inputs), in ExecuteNode()
114 for (TensorHandle* h : inputs_) { in Run()
124 return EagerKernelExecute(ctx_, inputs_, remote_func_params_, kernel_, in Run()
139 const absl::InlinedVector<TensorHandle*, 4>& inputs_; variable
160 inputs_(inputs), in AsyncExecuteNode()
175 for (auto handle : inputs_) { in AsyncExecuteNode()
185 for (auto handle : inputs_) { in ~AsyncExecuteNode()
192 for (TensorHandle* h : inputs_) { in Run()
203 ctx_, inputs_, remote_func_params_, kernel_, graph_collector_, in Run()
234 absl::InlinedVector<TensorHandle*, 4> inputs_; variable
Deager_operation.cc35 for (ImmediateExecutionTensorHandle* h : inputs_) { in Clear()
38 inputs_.clear(); in Clear()
244 reinterpret_cast<ImmediateExecutionTensorHandle* const*>(inputs_.data()), in GetInputs()
245 inputs_.size()); in GetInputs()
295 if (index >= inputs_.size()) { in SetInput()
297 inputs_.size()); in SetInput()
299 auto* previous = inputs_[index]; in SetInput()
307 inputs_[index] = input; in SetInput()
316 DCHECK(inputs_.empty()); in Reset()
408 dtypes[i] = inputs_[start + i]->DataType(); in InferInputListAttrs()
[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/core/distributed_runtime/eager/
Dremote_execute_node.h56 inputs_(inputs) { in RemoteExecuteNode()
66 for (auto handle : inputs_) { in RemoteExecuteNode()
72 for (const TensorHandle* input : inputs_) { in RemoteExecuteNode()
87 for (auto handle : inputs_) { in ~RemoteExecuteNode()
94 return RunShapeInference(ndef_, *lib_def_, inputs_, retvals_); in Prepare()
134 gtl::InlinedVector<TensorHandle*, 4> inputs_; variable
/external/tensorflow/tensorflow/core/kernels/
Dops_testutil.cc169 params_->inputs = &inputs_; in RunOpKernel()
192 CHECK_LT(input_index, inputs_.size()); in mutable_input()
193 return inputs_[input_index]; in mutable_input()
226 CHECK_GT(input_types_.size(), inputs_.size()) in AddInput()
228 bool is_ref = IsRefType(input_types_[inputs_.size()]); in AddInput()
232 CHECK_EQ(RemoveRefType(input_types_[inputs_.size()]), dtype); in AddInput()
233 inputs_.push_back({&lock_for_refs_, input}); in AddInput()
235 CHECK_EQ(input_types_[inputs_.size()], dtype); in AddInput()
236 inputs_.push_back({nullptr, input}); in AddInput()
253 inputs_.push_back({nullptr, input}); in AddResourceInputInternal()
Dquantize_and_dequantize_op_test.cc67 EXPECT_EQ(inputs_[1]->scalar<float>()(), 0.0); in TEST_F()
68 EXPECT_EQ(inputs_[2]->scalar<float>()(), 0.0); in TEST_F()
93 EXPECT_EQ(inputs_[1]->scalar<float>()(), 0.0); in TEST_F()
94 EXPECT_EQ(inputs_[2]->scalar<float>()(), 0.0); in TEST_F()
165 EXPECT_EQ(inputs_[1]->flat<float>()(slice_idx), 0.0); in TEST_P()
166 EXPECT_EQ(inputs_[2]->flat<float>()(slice_idx), 0.0); in TEST_P()
217 EXPECT_EQ(inputs_[1]->flat<float>()(slice_idx), 0.0); in TEST_P()
218 EXPECT_EQ(inputs_[2]->flat<float>()(slice_idx), 0.0); in TEST_P()
271 EXPECT_EQ(inputs_[1]->flat<float>()(slice_idx), 0.0); in TEST_P()
272 EXPECT_EQ(inputs_[2]->flat<float>()(slice_idx), 0.0); in TEST_P()
[all …]
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc79 inputs_.push_back(index); in OpInputs()
81 forwardable_.resize(inputs_.size()); in OpInputs()
85 int Size() const { return inputs_.size(); } in Size()
87 int TfLiteIndex(int i) const { return inputs_[i]; } in TfLiteIndex()
94 for (int i : inputs_) { in InitializeTensorSources()
111 std::vector<int> inputs_; member in tflite::flex::OpInputs
190 : inputs_(inputs), outputs_(outputs) {} in OpNode()
206 const OpInputs& inputs() const { return inputs_; } in inputs()
207 OpInputs* mutable_inputs() { return &inputs_; } in mutable_inputs()
212 int NumInputs() const { return inputs_.Size(); } in NumInputs()
[all …]
/external/tensorflow/tensorflow/c/
Dwhile_loop_test.cc36 DCHECK(inputs_.empty()); in Init()
43 inputs_.push_back({placeholder, 0}); in Init()
49 TF_NewWhile(graph_, &inputs_[0], inputs_.size(), s_))); in Init()
80 DCHECK_EQ(inputs_.size(), input_values.size()); in Run()
81 std::vector<std::pair<TF_Operation*, TF_Tensor*>> inputs(inputs_.size()); in Run()
84 inputs[i] = {inputs_[i].oper, Int32Tensor(v)}; in Run()
127 std::vector<TF_Output> inputs_; // The inputs to the while loop member in __anonbc36f9730111::CApiWhileLoopTest
315 params_->cond_output = inputs_[0]; in TEST_F()
356 params_->body_outputs[0] = inputs_[0]; in TEST_F()
388 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/lite/micro/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::__anonfa7cfece0111::BaseAddNOpModel
/external/tensorflow/tensorflow/lite/kernels/
Dadd_n_test.cc37 inputs_.push_back(AddInput(inputs[i])); in BaseAddNOpModel()
38 input_shapes.push_back(GetShape(inputs_[i])); in BaseAddNOpModel()
47 int input(int i) { return inputs_[i]; } in input()
50 std::vector<int> inputs_; member in tflite::__anon47e9de050111::BaseAddNOpModel
/external/tensorflow/tensorflow/core/kernels/image/
Dresize_area_op_test.cc41 inputs_.clear(); in SetRandomImageInput()
44 bool is_ref = IsRefType(input_types_[inputs_.size()]); in SetRandomImageInput()
50 CHECK_EQ(RemoveRefType(input_types_[inputs_.size()]), in SetRandomImageInput()
52 inputs_.push_back({&lock_for_refs_, input}); in SetRandomImageInput()
54 CHECK_EQ(input_types_[inputs_.size()], DataTypeToEnum<float>::v()); in SetRandomImageInput()
55 inputs_.push_back({nullptr, input}); in SetRandomImageInput()
/external/tensorflow/tensorflow/lite/profiling/
Dprofile_summarizer_test.cc83 PopulateTensor(inputs_[0], {x}); in SetInputs()
84 PopulateTensor(inputs_[1], {y}); in SetInputs()
89 int inputs_[2]; member in tflite::profiling::__anon5b4e8a840111::SimpleOpModel
95 inputs_[0] = AddInput({TensorType_INT32, {1}}); in Init()
96 inputs_[1] = AddInput({TensorType_INT32, {1}}); in Init()
99 BuildInterpreter({GetShape(inputs_[0]), GetShape(inputs_[1])}); in Init()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dtest_util.cc52 inputs_.emplace_back(std::move(tensor)); in SingleOpModel()
63 if (index >= inputs_.size() || in PopulateTensor()
64 inputs_[index].shape.DimensionsProduct() != data.size()) { in PopulateTensor()
67 inputs_[index].data = std::move(data); in PopulateTensor()
85 for (const auto& input : inputs_) { in Invoke()
/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::__anon7e249d170111::WhileLoopTest
/external/libtextclassifier/native/lang_id/common/fel/
Dtask-context.cc27 auto it = inputs_.find(name); in GetInputPath()
28 if (it != inputs_.end()) { in GetInputPath()
36 inputs_[name] = path; in SetInputPath()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dapi.cc464 RETURN_IF_ERROR(LinkTensors(inputs, factory, &inputs_)); in Initialize()
469 return GetExternalDefinitions(inputs_); in inputs()
477 if (index < 0 || index >= inputs_.size()) { in GetInputObject()
480 *object = inputs_[index]->GetExternalObject(); in GetInputObject()
493 if (index < 0 || index >= inputs_.size()) { in SetInputObject()
496 return inputs_[index]->SetExternalObject(object); in SetInputObject()
507 if (index > inputs_.size()) { in CopyFromExternalInput()
511 return inputs_[index]->CopyFromExternalObject(); in CopyFromExternalInput()
528 for (int i = 0; i < inputs_.size(); i++) { in Run()
580 std::vector<std::unique_ptr<TensorTie>> inputs_; member in tflite::gpu::cl::__anon117e98fd0111::InferenceRunnerImpl
[all …]
/external/tensorflow/tensorflow/core/framework/
Dmodel.h165 while (inputs_.size() > 0) { in ~Node()
166 queue.push_back(inputs_.front()); in ~Node()
167 inputs_.pop_front(); in ~Node()
175 while (node->inputs_.size() > 0) { in ~Node()
176 queue.push_back(node->inputs_.front()); in ~Node()
177 node->inputs_.pop_front(); in ~Node()
188 inputs_.push_back(node); in add_input()
236 return inputs_; in inputs()
305 inputs_.remove(input); in remove_input()
435 for (auto& input : inputs_) { in num_inputs()
[all …]

1234