Home
last modified time | relevance | path

Searched refs:input_tensors_ (Results 1 – 8 of 8) 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.h130 return input_tensors_.data() + tagged_node.node_item->input_start; in GetInputTensors()
176 std::vector<Entry> input_tensors_; variable
/external/tensorflow/tensorflow/lite/testing/
Dtf_driver.cc115 input_tensors_.reserve(input_layer.size()); in TfDriver()
120 input_tensors_[input_layer[i]] = {}; in TfDriver()
160 input_tensors_[input_names_[id]] = in ReshapeTensor()
168 auto tensor = input_tensors_[input_names_[id]]; in ResetTensor()
195 input_tensors_[input_names_[id]] = tensor; in Invoke()
197 auto status = session_->Run({input_tensors_.begin(), input_tensors_.end()}, in Invoke()
Dtf_driver.h78 std::unordered_map<string, tensorflow::Tensor> input_tensors_; variable
/external/tensorflow/tensorflow/compiler/tf2tensorrt/
Dtrt_convert_api_test.cc46 input_tensors_ = GetInputTensors(); in TrtConverterTest()
233 for (const std::vector<Tensor>& input : input_tensors_) { in RunAndCompareResults()
253 meta_graph_def.graph_def(), {"input"}, {"output"}, input_tensors_, in ConvertAndRunFrozenGraph()
273 &bundle, "serving_default", input_tensors_, param_.conv_params); in ConvertAndRunSavedModel()
284 std::vector<std::vector<Tensor>> input_tensors_; member in tensorflow::tensorrt::TrtConverterTest
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc176 input_tensors_ = input_tensors; in PreInputInit()
229 CHECK_LE(input_tensors_.size(), inputs_.size()); in PostInputInit()
230 input_tensors_.resize(inputs_.size()); in PostInputInit()
1170 const int input_tensors_size = input_tensors_.size(); in AttachContext()
1178 input_tensors_[i] != nullptr) { in AttachContext()
1181 input_tensors_[i]->SummarizeValue(256 /* max_values */), ">")); in AttachContext()
Dshape_inference.h348 return input_tensors_[idx]; in input_tensor()
379 input_tensors_ = input_tensors; in set_input_tensors()
822 std::vector<const Tensor*> input_tensors_; variable
/external/tensorflow/tensorflow/core/data/
Ddataset_test_base.h532 input_tensors_(std::move(input_tensors)), in TestDataset()
546 std::vector<std::unique_ptr<Tensor>> input_tensors_; variable