Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/lite/
Darena_planner.cc77 for (int tensor_index : graph_info_->outputs()) { in PlanAllocations() local
86 int tensor_index = node_inputs->data[j]; in PlanAllocations() local
94 for (int tensor_index : graph_info_->inputs()) { in PlanAllocations() local
107 int tensor_index = node_outputs->data[j]; in PlanAllocations() local
115 int tensor_index = node_inputs->data[j]; in PlanAllocations() local
185 TfLiteStatus ArenaPlanner::ResolveTensorAllocation(int tensor_index) { in ResolveTensorAllocation()
202 TfLiteStatus ArenaPlanner::CalculateTensorAllocation(int tensor_index) { in CalculateTensorAllocation()
217 TfLiteStatus ArenaPlanner::CalculateTensorDeallocation(int tensor_index) { in CalculateTensorDeallocation()
231 int tensor_index = node_temporaries->data[i]; in CalculateAllocationOfInternalTensors() local
244 int tensor_index = node_temporaries->data[i]; in CalculateDeallocationOfInternalTensors() local
Dinterpreter.h185 TfLiteTensor* tensor(int tensor_index) { in tensor()
203 T* typed_tensor(int tensor_index) { in typed_tensor()
Doptional_debug_tools.cc80 for (int tensor_index = 0; tensor_index < interpreter->tensors_size(); in PrintInterpreterState() local
Darena_planner_test.cc160 int64_t GetOffset(int tensor_index) { in GetOffset()
167 int64_t GetOffsetAfter(int tensor_index) { in GetOffsetAfter()
375 auto is_unallocated = [&](int tensor_index) { in TEST_F()
Dinterpreter.cc307 TfLiteStatus Interpreter::ResizeInputTensor(int tensor_index, in ResizeInputTensor()
487 int tensor_index, TfLiteType type, const char* name, in SetTensorParametersReadOnly()
513 int tensor_index, TfLiteType type, const char* name, in SetTensorParametersReadWrite()
Dinterpreter_test.cc640 for (int tensor_index = 0; tensor_index < 4; tensor_index++) { in SetUp() local
/external/tensorflow/tensorflow/python/layers/
Dnetwork.py388 tensor_index): argument
/external/tensorflow/tensorflow/core/kernels/data/
Dcache_dataset_ops.cc144 size_t tensor_index = 0; in GetNextInternal() local
/external/tensorflow/tensorflow/core/kernels/
Dtensor_array_ops.cc392 const Tensor* tensor_index; in Compute() local
477 const Tensor* tensor_index; in Compute() local
Dconv_ops_gpu_3.cu.cc164 Index<IndexCount> tensor_index; in FlatToTensorIndex() local
/external/tensorflow/tensorflow/core/debug/
Ddebug_grpc_io_utils_test.cc250 int tensor_index; in TEST_F() local