| /external/tensorflow/tensorflow/contrib/lite/ |
| arena_planner.cc | 77 for (int tensor_index : graph_info_->outputs()) {
78 refcounts[tensor_index]++;
86 int tensor_index = node_inputs->data[j]; local
87 if (tensor_index != kOptionalTensor) {
88 refcounts[tensor_index]++;
94 for (int tensor_index : graph_info_->inputs()) {
95 if (tensor_index != kOptionalTensor) {
96 alloc_queue_.push_back({0, tensor_index, AllocationInfo::ALLOC});
107 int tensor_index = node_outputs->data[j]; local
108 alloc_queue_.push_back({i, tensor_index, AllocationInfo::ALLOC})
115 int tensor_index = node_inputs->data[j]; local
231 int tensor_index = node_temporaries->data[i]; local
244 int tensor_index = node_temporaries->data[i]; local
[all...] |
| /external/tensorflow/tensorflow/core/debug/ |
| debug_grpc_io_utils_test.cc | 250 int tensor_index; local
251 strings::safe_strto32(items[2], &tensor_index);
255 ASSERT_EQ(tensor_index * tensor_index,
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| cache_dataset_ops.cc | 93 string FormatName(size_t item_index, size_t tensor_index) const {
95 tensor_index);
144 size_t tensor_index = 0; variable
146 DCHECK_LT(tensor_index, dataset()->num_tensors_);
147 string key = dataset()->FormatName(cur_index_, tensor_index++);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| conv_ops_gpu_3.cu.cc | 164 Index<IndexCount> tensor_index; local
167 tensor_index[i] = index - dims[i] * new_index;
170 return tensor_index;
[all...] |
| tensor_array_ops.cc | 392 const Tensor* tensor_index; variable
394 OP_REQUIRES_OK(ctx, ctx->input("index", &tensor_index));
397 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(tensor_index->shape()),
400 tensor_index->shape().DebugString()));
405 const int32 index = tensor_index->scalar<int32>()();
477 const Tensor* tensor_index; variable
478 OP_REQUIRES_OK(ctx, ctx->input("index", &tensor_index));
480 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(tensor_index->shape()),
483 tensor_index->shape().DebugString()));
489 const int32 index = tensor_index->scalar<int32>()()
[all...] |
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorConvolution.h | 593 const int tensor_index = plane_input_offset + indexMapper.mapCudaInputKernelToTensorInputOffset(i+first_x); local
594 s[i + plane_kernel_offset] = eval.coeff(tensor_index);
610 const int tensor_index = plane_output_offset + indexMapper.mapCudaOutputKernelToTensorOutputOffset(i+first_x); local
611 buffer[tensor_index] = result;
652 const int tensor_index = plane_input_offset + indexMapper.mapCudaInputKernelToTensorInputOffset(i+first_x, j+first_y); local
653 s[i + input_offset] = eval.coeff(tensor_index);
676 const int tensor_index = plane_output_offset + indexMapper.mapCudaOutputKernelToTensorOutputOffset(i+first_x, j+first_y); local
677 buffer[tensor_index] = result;
717 const int tensor_index = plane_input_offset + indexMapper.mapCudaInputKernelToTensorInputOffset(i+first_x, j+first_y, k+first_z); local
718 s[i + num_x_input * (j + num_y_input * (k + plane_kernel_offset))] = eval.coeff(tensor_index);
742 const int tensor_index = plane_output_offset + indexMapper.mapCudaOutputKernelToTensorOutputOffset(i+first_x, j+first_y, k+first_z); local
[all...] |
