Home
last modified time | relevance | path

Searched refs:input_tensors_ (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dsimple_propagator_state.cc36 input_tensors_(finfo.total_inputs), in SimplePropagatorState()
87 input_tensors_[dst_loc] = std::move((*outputs)[src_slot]); in PropagateOutputs()
89 input_tensors_[dst_loc] = (*outputs)[src_slot]; in PropagateOutputs()
111 DumpPendingNodeState(*node, input_tensors_.data(), false); in DumpState()
117 DumpActiveNodeState(*node, input_tensors_.data()); in DumpState()
122 for (size_t i = 0; i < input_tensors_.size(); ++i) { in DumpState()
123 const Entry& input = input_tensors_[i]; in DumpState()
Dsimple_propagator_state.h129 return input_tensors_.data() + tagged_node.node_item->input_start; in GetInputTensors()
175 std::vector<Entry> input_tensors_; variable
/external/tensorflow/tensorflow/lite/testing/
Dtf_driver.cc114 input_tensors_.reserve(input_layer.size()); in TfDriver()
119 input_tensors_[input_layer[i]] = {}; in TfDriver()
195 input_tensors_[input_names_[id]] = tensor; in SetInput()
200 auto tensor = input_tensors_[input_names_[id]]; in ResetTensor()
220 input_tensors_[input_names_[id]] = in ReshapeTensor()
256 auto status = session_->Run({input_tensors_.begin(), input_tensors_.end()}, in Invoke()
Dtf_driver.h71 std::unordered_map<string, tensorflow::Tensor> input_tensors_; variable
/external/tensorflow/tensorflow/lite/micro/
Dmicro_interpreter.cc122 input_tensors_(nullptr), in MicroInterpreter()
140 input_tensors_(nullptr), in MicroInterpreter()
255 input_tensors_ = in AllocateTensors()
258 if (input_tensors_ == nullptr) { in AllocateTensors()
268 input_tensors_[i] = allocator_.AllocatePersistentTfLiteTensor( in AllocateTensors()
270 if (input_tensors_[i] == nullptr) { in AllocateTensors()
364 return input_tensors_[index]; in input()
Dmicro_interpreter.h204 TfLiteTensor** input_tensors_; variable
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils_test.cc92 input_tensors_, /*dry_run_inference*/ true, &graph_def_with_shapetype)); in FuseByInOut()
129 input_tensors_, /*dry_run_inference=*/true, &graph_def_); in BuildAndAddTensorShape()
143 graph_def_, input_tensors_, &graph_def_); in FuseByPlacedArguments()
150 input_tensors_); in IsFuseReady()
163 const std::vector<std::pair<string, Tensor>> input_tensors_{ member in tensorflow::__anon388f233f0111::FuseRemoteGraphMultipleAddOpsTest
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.h272 return input_tensors_[idx]; in input_tensor()
303 input_tensors_ = input_tensors; in set_input_tensors()
736 std::vector<const Tensor*> input_tensors_; variable
Dshape_inference.cc169 input_tensors_ = input_tensors; in PreInputInit()
222 CHECK_LE(input_tensors_.size(), inputs_.size()); in PostInputInit()
223 input_tensors_.resize(inputs_.size()); in PostInputInit()
1137 const int input_tensors_size = input_tensors_.size(); in AttachContext()
1145 input_tensors_[i] != nullptr) { in AttachContext()
1148 input_tensors_[i]->SummarizeValue(256 /* max_values */), ">")); in AttachContext()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.h486 input_tensors_(std::move(input_tensors)), in TestDataset()
500 std::vector<std::unique_ptr<Tensor>> input_tensors_; variable