Home
last modified time | relevance | path

Searched refs:tensor_index (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/
Darena_planner.cc77 for (int tensor_index : graph_info_->outputs()) { in PlanAllocations() local
78 refcounts[tensor_index]++; in PlanAllocations()
86 int tensor_index = node_inputs->data[j]; in PlanAllocations() local
87 if (tensor_index != kOptionalTensor) { in PlanAllocations()
88 refcounts[tensor_index]++; in PlanAllocations()
94 for (int tensor_index : graph_info_->inputs()) { in PlanAllocations() local
95 if (tensor_index != kOptionalTensor) { in PlanAllocations()
96 alloc_queue_.push_back({0, tensor_index, AllocationInfo::ALLOC}); in PlanAllocations()
107 int tensor_index = node_outputs->data[j]; in PlanAllocations() local
108 alloc_queue_.push_back({i, tensor_index, AllocationInfo::ALLOC}); in PlanAllocations()
[all …]
Dinterpreter.h137 int tensor_index, TfLiteType type, const char* name,
146 int tensor_index, TfLiteType type, const char* name,
185 TfLiteTensor* tensor(int tensor_index) { in tensor() argument
186 if (tensor_index >= context_.tensors_size || tensor_index < 0) in tensor()
188 return &context_.tensors[tensor_index]; in tensor()
203 T* typed_tensor(int tensor_index) { in typed_tensor() argument
204 if (TfLiteTensor* tensor_ptr = tensor(tensor_index)) { in typed_tensor()
232 TfLiteStatus ResizeInputTensor(int tensor_index,
Doptional_debug_tools.cc80 for (int tensor_index = 0; tensor_index < interpreter->tensors_size(); in PrintInterpreterState() local
81 tensor_index++) { in PrintInterpreterState()
82 TfLiteTensor* tensor = interpreter->tensor(tensor_index); in PrintInterpreterState()
83 printf("Tensor %3d %10s %15s %10zu bytes (%4.1f MB) ", tensor_index, in PrintInterpreterState()
Dinterpreter.cc307 TfLiteStatus Interpreter::ResizeInputTensor(int tensor_index, in ResizeInputTensor() argument
313 tensor_index < context_.tensors_size && tensor_index >= 0); in ResizeInputTensor()
316 return ResizeTensorImpl(&context_.tensors[tensor_index], dims_lite); in ResizeInputTensor()
487 int tensor_index, TfLiteType type, const char* name, in SetTensorParametersReadOnly() argument
491 tensor_index < context_.tensors_size && tensor_index >= 0); in SetTensorParametersReadOnly()
504 kTfLiteMmapRo, allocation, &context_.tensors[tensor_index]); in SetTensorParametersReadOnly()
513 int tensor_index, TfLiteType type, const char* name, in SetTensorParametersReadWrite() argument
517 tensor_index < context_.tensors_size && tensor_index >= 0); in SetTensorParametersReadWrite()
531 nullptr, &context_.tensors[tensor_index]); in SetTensorParametersReadWrite()
Darena_planner.h70 TfLiteStatus ResolveTensorAllocation(int tensor_index);
73 TfLiteStatus CalculateTensorAllocation(int tensor_index);
76 TfLiteStatus CalculateTensorDeallocation(int tensor_index);
Darena_planner_test.cc160 int64_t GetOffset(int tensor_index) { in GetOffset() argument
161 const TfLiteTensor& tensor = (*graph_->tensors())[tensor_index]; in GetOffset()
167 int64_t GetOffsetAfter(int tensor_index) { in GetOffsetAfter() argument
168 const TfLiteTensor& tensor = (*graph_->tensors())[tensor_index]; in GetOffsetAfter()
169 int64_t offset = GetOffset(tensor_index) + tensor.bytes; in GetOffsetAfter()
375 auto is_unallocated = [&](int tensor_index) { in TEST_F() argument
376 return (*graph.tensors())[tensor_index].data.raw == nullptr; in TEST_F()
Dinterpreter_test.cc640 for (int tensor_index = 0; tensor_index < 4; tensor_index++) { in SetUp() local
642 tensor_index, kTfLiteFloat32, "", {3}, quantized), in SetUp()
/external/tensorflow/tensorflow/contrib/lite/tools/
Dvisualize.py213 for tensor_input_position, tensor_index in enumerate(op["inputs"]):
214 if tensor_index not in first:
215 first[tensor_index] = (
219 "source": TensorName(tensor_index),
222 for tensor_index in op["outputs"]:
224 "target": TensorName(tensor_index),
234 for tensor_index, tensor in enumerate(g["tensors"]):
236 first[tensor_index] if tensor_index in first else len(g["operators"]))
239 "id": TensorName(tensor_index),
240 "name": "%s (%d)" % (tensor["name"], tensor_index),
/external/tensorflow/tensorflow/python/layers/
Dnetwork.py336 layer, node_index, tensor_index = x._keras_history
360 layer, node_index, tensor_index = x._keras_history # pylint: disable=protected-access
362 self._output_coordinates.append((layer, node_index, tensor_index))
366 layer, node_index, tensor_index = x._keras_history # pylint: disable=protected-access
370 assert tensor_index == 0
372 self._input_coordinates.append((layer, node_index, tensor_index))
388 tensor_index): argument
434 tensor_index = node.tensor_indices[i]
436 node_index, tensor_index)
445 layer, node_index, tensor_index = x._keras_history # pylint: disable=protected-access
[all …]
Dbase.py838 inbound_layer, node_index, tensor_index = x._keras_history # pylint: disable=protected-access
841 tensor_indices.append(tensor_index)
/external/tensorflow/tensorflow/core/kernels/data/
Dcache_dataset_ops.cc93 string FormatName(size_t item_index, size_t tensor_index) const { in FormatName()
95 tensor_index); in FormatName()
144 size_t tensor_index = 0; in GetNextInternal() local
146 DCHECK_LT(tensor_index, dataset()->num_tensors_); in GetNextInternal()
147 string key = dataset()->FormatName(cur_index_, tensor_index++); in GetNextInternal()
/external/tensorflow/tensorflow/python/keras/_impl/keras/engine/
Dtopology.py929 tensor_index = node.tensor_indices[i]
934 [inbound_layer.name, new_node_index, tensor_index, kwargs])
947 layer, node_index, tensor_index = self._input_coordinates[i]
953 model_inputs.append([layer.name, new_node_index, tensor_index])
957 layer, node_index, tensor_index = self._output_coordinates[i]
963 model_outputs.append([layer.name, new_node_index, tensor_index])
1082 layer_name, node_index, tensor_index = layer_data
1086 input_tensors.append(layer_output_tensors[tensor_index])
1088 layer_name, node_index, tensor_index = layer_data
1092 output_tensors.append(layer_output_tensors[tensor_index])
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorConvolution.h593 …const int tensor_index = plane_input_offset + indexMapper.mapCudaInputKernelToTensorInputOffset(i+…
594 s[i + plane_kernel_offset] = eval.coeff(tensor_index);
610 …const int tensor_index = plane_output_offset + indexMapper.mapCudaOutputKernelToTensorOutputOffset…
611 buffer[tensor_index] = result;
652 …const int tensor_index = plane_input_offset + indexMapper.mapCudaInputKernelToTensorInputOffset(i+…
653 s[i + input_offset] = eval.coeff(tensor_index);
676 …const int tensor_index = plane_output_offset + indexMapper.mapCudaOutputKernelToTensorOutputOffset…
677 buffer[tensor_index] = result;
717 …const int tensor_index = plane_input_offset + indexMapper.mapCudaInputKernelToTensorInputOffset(i+…
718 … s[i + num_x_input * (j + num_y_input * (k + plane_kernel_offset))] = eval.coeff(tensor_index);
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dtensor_array_ops.cc392 const Tensor* tensor_index; in Compute() local
394 OP_REQUIRES_OK(ctx, ctx->input("index", &tensor_index)); in Compute()
397 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(tensor_index->shape()), in Compute()
400 tensor_index->shape().DebugString())); in Compute()
405 const int32 index = tensor_index->scalar<int32>()(); in Compute()
477 const Tensor* tensor_index; in Compute() local
478 OP_REQUIRES_OK(ctx, ctx->input("index", &tensor_index)); in Compute()
480 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(tensor_index->shape()), in Compute()
483 tensor_index->shape().DebugString())); in Compute()
489 const int32 index = tensor_index->scalar<int32>()(); in Compute()
Dconv_ops_gpu_3.cu.cc164 Index<IndexCount> tensor_index; in FlatToTensorIndex() local
167 tensor_index[i] = index - dims[i] * new_index; in FlatToTensorIndex()
170 return tensor_index; in FlatToTensorIndex()
/external/tensorflow/tensorflow/core/debug/
Ddebug_grpc_io_utils_test.cc250 int tensor_index; in TEST_F() local
251 strings::safe_strto32(items[2], &tensor_index); in TEST_F()
255 ASSERT_EQ(tensor_index * tensor_index, in TEST_F()
/external/tensorflow/tensorflow/contrib/lite/g3doc/
Dapis.md125 TfLiteStatus ResizeInputTensor(int tensor_index,