Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/
Darena_planner.cc116 for (int tensor_index : graph_info_->outputs()) { in PlanAllocations() local
122 for (int tensor_index : graph_info_->variables()) { in PlanAllocations() local
135 for (int tensor_index : graph_info_->inputs()) { in PlanAllocations() local
147 int tensor_index = node_inputs->data[j]; in PlanAllocations() local
161 int tensor_index = node_outputs->data[j]; in PlanAllocations() local
170 int tensor_index = node_inputs->data[j]; in PlanAllocations() local
200 int tensor_index = node_temporaries->data[j]; in ExecuteAllocations() local
311 for (const auto& tensor_index : tensor_order) { in CalculateAllocations() local
320 for (const auto& tensor_index : tensor_order) { in CalculateAllocations() local
341 TfLiteStatus ArenaPlanner::ResolveTensorAllocation(int tensor_index) { in ResolveTensorAllocation()
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()
Dinterpreter.cc209 TfLiteStatus Interpreter::ResizeInputTensor(int tensor_index, in ResizeInputTensor()
215 int tensor_index, const std::vector<int>& dims) { in ResizeInputTensorStrict()
232 for (int tensor_index : outputs()) { in Invoke() local
248 int tensor_index, TfLiteType type, const char* name, in SetTensorParametersReadOnly()
257 int tensor_index, TfLiteType type, const char* name, in SetTensorParametersReadWrite()
266 int tensor_index, TfLiteType type, const char* name, const size_t rank, in SetTensorParametersReadOnly()
276 int tensor_index, TfLiteType type, const char* name, const size_t rank, in SetTensorParametersReadWrite()
Dinterpreter.h250 TfLiteTensor* tensor(int tensor_index) { in tensor()
255 const TfLiteTensor* tensor(int tensor_index) const { in tensor()
276 T* typed_tensor(int tensor_index) { in typed_tensor()
288 const T* typed_tensor(int tensor_index) const { in typed_tensor()
370 const int tensor_index = GetTensorIndexFromSignature( in input_tensor_by_signature() local
384 const int tensor_index = GetTensorIndexFromSignature( in output_tensor_by_signature() local
561 TfLiteStatus EnsureTensorDataIsReadable(int tensor_index) { in EnsureTensorDataIsReadable()
Dinterpreter_experimental.cc40 int tensor_index, const TfLiteCustomAllocation& allocation, int64_t flags) { in SetCustomAllocationForTensor()
104 TfLiteStatus Interpreter::SetBufferHandle(int tensor_index, in SetBufferHandle()
123 TfLiteStatus Interpreter::GetBufferHandle(int tensor_index, in GetBufferHandle()
Doptional_debug_tools.cc51 void Update(size_t tensor_index, const TfLiteTensor& tensor) { in Update()
154 void Update(size_t tensor_index, const TfLiteTensor& tensor) { in Update()
364 for (size_t tensor_index = 0; tensor_index < subgraph.tensors_size(); in PrintInterpreterState() local
373 for (size_t tensor_index = 0; tensor_index < subgraph.tensors_size(); in PrintInterpreterState() local
Dsignature_runner.cc68 for (int tensor_index : subgraph_->outputs()) { in Invoke() local
/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()
95 void FlexModelTest::SetConstTensor(int tensor_index, in SetConstTensor()
Dbuffer_map.cc46 void BufferMap::SetFromTfLite(int tensor_index, const TfLiteTensor* tensor) { in SetFromTfLite()
56 void BufferMap::SetFromTensorFlow(int tensor_index, tensorflow::Tensor tensor) { in SetFromTensorFlow()
Dkernel.cc393 for (auto tensor_index : TfLiteIntArrayView(params->output_tensors)) { in Init() local
399 for (auto tensor_index : TfLiteIntArrayView(params->input_tensors)) { in Init() local
463 for (auto tensor_index : op_data_->subgraph_inputs) { in Prepare() local
489 for (auto tensor_index : op_data_->subgraph_outputs) { in Prepare() local
606 for (auto tensor_index : op_data_->subgraph_inputs) { in Eval() local
634 for (auto tensor_index : op_data_->subgraph_outputs) { in Eval() local
/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util.cc43 int tensor_index) { in GetTensorAtIndex()
55 int* tensor_index) { in ValidateTensorIndexingSafe()
78 const int tensor_index = tensor_indices[index]; in ValidateTensorIndexing() local
88 const int tensor_index = ValidateTensorIndexing( in GetMutableInput() local
99 int tensor_index; in GetMutableInputSafe() local
128 const int tensor_index = ValidateTensorIndexing( in GetOutput() local
138 int tensor_index; in GetOutputSafe() local
154 const int tensor_index = ValidateTensorIndexing( in GetTemporary() local
165 int tensor_index; in GetTemporarySafe() local
175 const int tensor_index = ValidateTensorIndexing( in GetIntermediates() local
[all …]
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dhexagon_delegate_kernel.cc98 const auto tensor_index = node->inputs->data[input_idx]; in Eval() local
124 for (auto tensor_index : TfLiteIntArrayView(node->outputs)) { in Eval() local
211 for (auto tensor_index : TfLiteIntArrayView(node->inputs)) { in Prepare() local
214 for (auto tensor_index : TfLiteIntArrayView(node->outputs)) { in Prepare() local
217 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/python/
Dinterpreter.py511 def _get_tensor_details(self, tensor_index): argument
624 def set_tensor(self, tensor_index, value): argument
780 def get_tensor(self, tensor_index): argument
795 def tensor(self, tensor_index): argument
/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/tools/serialization/
Dwriter_lib.cc170 for (auto tensor_index : in ExportTensors() local
177 for (int tensor_index = 0; tensor_index < subgraph_->tensors_size(); in ExportTensors() local
186 for (int tensor_index = 0; tensor_index < subgraph_->tensors_size(); in ExportTensors() local
348 for (int tensor_index : TfLiteIntArrayView(node.inputs)) { in CheckInputOutput() local
377 for (int tensor_index : outputs) { in CheckInputOutput() local
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h144 TfLiteTensor* tensor(int tensor_index) { in tensor()
153 const TfLiteTensor* tensor(int tensor_index) const { in tensor()
282 TfLiteStatus EnsureTensorDataIsReadable(int tensor_index) { in EnsureTensorDataIsReadable()
Dsubgraph.cc435 for (int tensor_index : node_subset.output_tensors) { in ReplaceNodeSubsetsWithDelegateKernels() local
855 int tensor_index = tensor_indexes->data[i]; in AnyTensorOfTypeResource() local
880 TfLiteStatus Subgraph::ResizeInputTensor(int tensor_index, in ResizeInputTensor()
912 TfLiteStatus Subgraph::ResizeInputTensorStrict(int tensor_index, in ResizeInputTensorStrict()
1116 int tensor_index = node.inputs->data[i]; in Invoke() local
1280 int tensor_index, TfLiteType type, const char* name, const size_t rank, in SetTensorParametersReadOnly()
1339 int tensor_index, TfLiteType type, const char* name, const size_t rank, in SetTensorParametersReadWrite()
1709 int tensor_index, const TfLiteCustomAllocation& allocation, int64_t flags) { in SetCustomAllocationForTensor()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibration_logger.h52 TfLiteStatus LogTensorValue(int subgraph_index, int tensor_index, in LogTensorValue()
Dcalibration_reader.cc42 int subgraph_index, tensor_index; in AddCalibrationToModel() local
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc903 int tensor_index, int node_index) { in CheckTensorType()
915 int tensor_index, int node_index) { in CheckTensorFloat32Type()
922 int tensor_index, in CheckTensorFloat32OrQInt8Type()
956 int tensor_index, in CheckTensorFloat32OrQUInt8Type()
1010 int expected_quantized_dimension, int tensor_index, int node_index) { in CheckTensorFloat32OrQCInt8Type()
1077 int tensor_index, in CheckTensorFloat32OrQInt32Type()
1112 TfLiteContext* context, const TfLiteTensor& tensor, int tensor_index, in CheckTensorFloat32OrQCInt32Type()
1143 int tensor_index) { in CheckTensorShape()
1186 int tensor_index) { in CheckTensorShape()
1193 int tensor_index, int node_index) { in CheckSlopeTensorShape()
[all …]
/external/tensorflow/tensorflow/core/protobuf/tpu/
Dtpu_embedding_output_layout.proto26 int32 tensor_index = 1; field
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_quantize_helper.h71 inline QuantizedType GetIntermediateElementType(LstmOp op, int tensor_index) { in GetIntermediateElementType()
530 for (int tensor_index : in GetLstmOpQuantSpec() local
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dinterpreter_wrapper_pybind11.cc163 int subgraph_index) { in PYBIND11_MODULE()
194 int tensor_index, int subgraph_index) { in PYBIND11_MODULE()

123