Home
last modified time | relevance | path

Searched defs:tensor_index (Results 1 – 25 of 57) sorted by relevance

123

/external/tensorflow/tensorflow/lite/
Dsimple_planner.cc102 for (int tensor_index : graph_info_->outputs()) { in PlanAllocations() local
108 for (int tensor_index : graph_info_->variables()) { in PlanAllocations() local
121 for (int tensor_index : graph_info_->inputs()) { in PlanAllocations() local
133 int tensor_index = node_inputs->data[j]; in PlanAllocations() local
147 int tensor_index = node_outputs->data[j]; in PlanAllocations() local
155 int tensor_index = node_inputs->data[j]; in PlanAllocations() local
181 int tensor_index = node_temporaries->data[j]; in ExecuteAllocations() local
234 TfLiteStatus SimplePlanner::ResolveTensorAllocation(int tensor_index) { in ResolveTensorAllocation()
Darena_planner.cc120 for (int tensor_index : graph_info_->outputs()) { in PlanAllocations() local
126 for (int tensor_index : graph_info_->variables()) { in PlanAllocations() local
139 for (int tensor_index : graph_info_->inputs()) { in PlanAllocations() local
151 int tensor_index = node_inputs->data[j]; in PlanAllocations() local
165 int tensor_index = node_outputs->data[j]; in PlanAllocations() local
174 int tensor_index = node_inputs->data[j]; in PlanAllocations() local
204 int tensor_index = node_temporaries->data[j]; in ExecuteAllocations() local
317 for (const auto& tensor_index : tensor_order) { in CalculateAllocations() local
326 for (const auto& tensor_index : tensor_order) { in CalculateAllocations() local
347 TfLiteStatus ArenaPlanner::ResolveTensorAllocation(int tensor_index) { in ResolveTensorAllocation()
Dinterpreter.cc211 TfLiteStatus Interpreter::ResizeInputTensor(int tensor_index, in ResizeInputTensor()
217 int tensor_index, const std::vector<int>& dims) { in ResizeInputTensorStrict()
234 for (int tensor_index : outputs()) { in Invoke() local
250 int tensor_index, TfLiteType type, const char* name, in SetTensorParametersReadOnly()
259 int tensor_index, TfLiteType type, const char* name, in SetTensorParametersReadWrite()
268 int tensor_index, TfLiteType type, const char* name, const size_t rank, in SetTensorParametersReadOnly()
278 int tensor_index, TfLiteType type, const char* name, const size_t rank, in SetTensorParametersReadWrite()
Dinterpreter.h251 TfLiteTensor* tensor(int tensor_index) { in tensor()
256 const TfLiteTensor* tensor(int tensor_index) const { in tensor()
277 T* typed_tensor(int tensor_index) { in typed_tensor()
289 const T* typed_tensor(int tensor_index) const { in typed_tensor()
371 const int tensor_index = GetTensorIndexFromSignature( in input_tensor_by_signature() local
385 const int tensor_index = GetTensorIndexFromSignature( in output_tensor_by_signature() local
578 TfLiteStatus EnsureTensorDataIsReadable(int tensor_index) { in EnsureTensorDataIsReadable()
Dinterpreter_experimental.cc39 int tensor_index, const TfLiteCustomAllocation& allocation, int64_t flags) { in SetCustomAllocationForTensor()
81 TfLiteStatus Interpreter::SetBufferHandle(int tensor_index, in SetBufferHandle()
100 TfLiteStatus Interpreter::GetBufferHandle(int tensor_index, in GetBufferHandle()
Doptional_debug_tools.cc55 void Update(size_t tensor_index, const TfLiteTensor& tensor) { in Update()
153 void Update(size_t tensor_index, const TfLiteTensor& tensor) { in Update()
197 void Update(size_t tensor_index, const TfLiteTensor& tensor) { in Update()
412 for (size_t tensor_index = 0; tensor_index < subgraph.tensors_size(); in PrintInterpreterState() local
421 for (size_t tensor_index = 0; tensor_index < subgraph.tensors_size(); in PrintInterpreterState() local
Dsignature_runner.cc77 for (int tensor_index : subgraph_->outputs()) { in Invoke() local
Darena_planner_test.cc229 std::ptrdiff_t GetOffset(int tensor_index) { in GetOffset()
236 std::ptrdiff_t GetOffsetAfter(int tensor_index) { in GetOffsetAfter()
247 bool IsUnallocated(int tensor_index) { in IsUnallocated()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dtest_util.h52 void SetTypedValues(int tensor_index, const std::vector<T>& values) { in SetTypedValues()
59 std::vector<T> GetTypedValues(int tensor_index) { in GetTypedValues()
66 void SetValues(int tensor_index, const std::vector<float>& values) { in SetValues()
72 std::vector<float> GetValues(int tensor_index) { in GetValues()
Dtest_util.cc28 void FlexModelTest::SetStringValues(int tensor_index, in SetStringValues()
51 void FlexModelTest::SetShape(int tensor_index, const std::vector<int>& values) { in SetShape()
56 std::vector<int> FlexModelTest::GetShape(int tensor_index) { in GetShape()
66 TfLiteType FlexModelTest::GetType(int tensor_index) { in GetType()
70 bool FlexModelTest::IsDynamicTensor(int tensor_index) { in IsDynamicTensor()
90 void FlexModelTest::SetConstTensor(int tensor_index, in SetConstTensor()
Dbuffer_map.cc49 void BufferMap::SetFromTfLite(int tensor_index, const TfLiteTensor* tensor, in SetFromTfLite()
61 void BufferMap::SetFromTensorFlow(int tensor_index, tensorflow::Tensor tensor) { in SetFromTensorFlow()
Dkernel.cc300 int tensor_index, int node_index) { in ShouldPersistTensorflowTensor()
455 for (auto tensor_index : TfLiteIntArrayView(params->output_tensors)) { in Init() local
461 for (auto tensor_index : TfLiteIntArrayView(params->input_tensors)) { in Init() local
503 for (auto tensor_index : TfLiteIntArrayView(node->inputs)) { in Init() local
557 for (auto tensor_index : op_data_->subgraph_inputs) { in Prepare() local
583 for (auto tensor_index : op_data_->subgraph_outputs) { in Prepare() local
714 for (auto tensor_index : op_data_->subgraph_inputs) { in Eval() local
762 for (auto tensor_index : op_data_->subgraph_outputs) { in Eval() local
/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util.cc44 int tensor_index) { in GetTensorAtIndex()
56 int* tensor_index) { in ValidateTensorIndexingSafe()
79 const int tensor_index = tensor_indices[index]; in ValidateTensorIndexing() local
89 const int tensor_index = ValidateTensorIndexing( in GetMutableInput() local
100 int tensor_index; in GetMutableInputSafe() local
129 const int tensor_index = ValidateTensorIndexing( in GetOutput() local
139 int tensor_index; in GetOutputSafe() local
155 const int tensor_index = ValidateTensorIndexing( in GetTemporary() local
166 int tensor_index; in GetTemporarySafe() local
176 const int tensor_index = ValidateTensorIndexing( in GetIntermediates() local
[all …]
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dhexagon_delegate_kernel.cc100 const auto tensor_index = node->inputs->data[input_idx]; in Eval() local
126 for (auto tensor_index : TfLiteIntArrayView(node->outputs)) { in Eval() local
215 for (auto tensor_index : TfLiteIntArrayView(node->inputs)) { in Prepare() local
218 for (auto tensor_index : TfLiteIntArrayView(node->outputs)) { in Prepare() local
221 for (auto tensor_index : tensors) { in Prepare() local
/external/tensorflow/tensorflow/lite/delegates/gpu/
Dgl_delegate.cc80 int tensor_index; member
122 absl::Status BindBufferToTensor(GLuint ssbo, int tensor_index) { in BindBufferToTensor()
148 auto find_value = [&](int tensor_index) -> Value* { in Prepare()
172 const int tensor_index = delegate_params->input_tensors->data[i]; in Prepare() local
214 const int tensor_index = delegate_params->output_tensors->data[i]; in Prepare() local
482 int tensor_index) { in TfLiteGpuDelegateBindBufferToTensor()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dgpu_api_delegate.cc196 for (auto tensor_index : input_refs) { in Prepare() local
203 for (auto tensor_index : output_refs) { in Prepare() local
232 void BindGlBufferToTensor(GLuint buffer_id, int tensor_index, in BindGlBufferToTensor()
381 TfLiteDelegate* delegate, GLuint buffer_id, int tensor_index, in TfLiteGpuDelegateBindGlBufferToTensor()
/external/tensorflow/tensorflow/lite/python/
Dinterpreter.py568 def _get_tensor_details(self, tensor_index): argument
681 def set_tensor(self, tensor_index, value): argument
837 def get_tensor(self, tensor_index, subgraph_index=0): argument
854 def tensor(self, tensor_index): argument
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib.cc175 for (auto tensor_index : in ExportTensors() local
182 for (int tensor_index = 0; tensor_index < subgraph_->tensors_size(); in ExportTensors() local
191 for (int tensor_index = 0; tensor_index < subgraph_->tensors_size(); in ExportTensors() local
366 for (int tensor_index : TfLiteIntArrayView(node.inputs)) { in CheckInputOutput() local
395 for (int tensor_index : outputs) { in CheckInputOutput() local
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc453 for (int tensor_index : node_subset.output_tensors) { in ReplaceNodeSubsetsWithDelegateKernels() local
867 int tensor_index = tensor_indexes->data[i]; in AnyTensorOfTypeResource() local
892 TfLiteStatus Subgraph::ResizeInputTensor(int tensor_index, in ResizeInputTensor()
924 TfLiteStatus Subgraph::ResizeInputTensorStrict(int tensor_index, in ResizeInputTensorStrict()
1064 int tensor_index = node->outputs->data[i]; in MayAllocateOpOutput() local
1197 for (int tensor_index : graph_info->variables()) { in RemoveUnusedInputs() local
1205 int tensor_index = node_inputs->data[j]; in RemoveUnusedInputs() local
1276 int tensor_index = node.inputs->data[i]; in Invoke() local
1451 int tensor_index, TfLiteType type, const char* name, const size_t ndims, in SetTensorParametersReadOnly()
1508 int tensor_index, TfLiteType type, const char* name, const size_t ndims, in SetTensorParametersReadWrite()
[all …]
Dsubgraph.h148 TfLiteTensor* tensor(int tensor_index) { in tensor()
157 const TfLiteTensor* tensor(int tensor_index) const { in tensor()
289 TfLiteStatus EnsureTensorDataIsReadable(int tensor_index) { in EnsureTensorDataIsReadable()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibration_logger.h52 TfLiteStatus LogTensorValue(int subgraph_index, int tensor_index, in LogTensorValue()
Dcalibration_reader.cc45 int subgraph_index, tensor_index; in AddCalibrationToModel() local
/external/tensorflow/tensorflow/dtensor/cc/
Dsave_restore_util.h32 int64_t tensor_index; member
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dinterpreter_wrapper_pybind11.cc168 int subgraph_index) { in PYBIND11_MODULE()
199 int tensor_index, int subgraph_index) { in PYBIND11_MODULE()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc1291 int tensor_index, int node_index) { in CheckTensorType()
1303 int tensor_index, int node_index) { in CheckTensorFloat32Type()
1311 int tensor_index, in CheckTensorFloat32OrQInt8Type()
1347 int tensor_index, in CheckTensorQInt8OrQUInt8Type()
1401 int tensor_index, in CheckTensorFloat32OrQUInt8Type()
1457 int tensor_index, int node_index) { in CheckTensorFloat32OrQCInt8Type()
1526 int tensor_index, in CheckTensorFloat32OrQInt32Type()
1564 const TfLiteTensor& tensor, int tensor_index, int node_index) { in CheckTensorFloat32OrQCInt32Type()
1595 int tensor_index) { in CheckTensorShape()
1638 int tensor_index) { in CheckTensorShape()
[all …]

123