Searched refs:tensor_indices (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/profiling/ |
| D | profile_summarizer.cc | 45 const TfLiteIntArray* tensor_indices) { in GetTensorNames() argument
47 tensors.reserve(tensor_indices->size); in GetTensorNames()
48 for (int i = 0; i < tensor_indices->size; i++) { in GetTensorNames()
49 tensors.push_back(GetTensorName(interpreter, tensor_indices->data[i])); in GetTensorNames()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_array_ops.cc | 622 const Tensor* tensor_indices; in Compute() local
623 OP_REQUIRES_OK(ctx, ctx->input("indices", &tensor_indices)); in Compute()
624 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(tensor_indices->shape()), in Compute()
627 tensor_indices->shape().DebugString())); in Compute()
628 const auto indices_t = tensor_indices->vec<int32>(); in Compute()
629 num_indices = tensor_indices->NumElements(); in Compute()
1068 const Tensor* tensor_indices; in Compute() local
1069 OP_REQUIRES_OK(ctx, ctx->input("indices", &tensor_indices)); in Compute()
1070 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(tensor_indices->shape()), in Compute()
1073 tensor_indices->shape().DebugString())); in Compute()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util.cc | 48 const int* tensor_indices, in ValidateTensorIndexingSafe() argument
56 if (tensor_indices[index] == kTfLiteOptionalTensor) { in ValidateTensorIndexingSafe()
63 *tensor_index = tensor_indices[index]; in ValidateTensorIndexingSafe()
70 int max_size, const int* tensor_indices) { in ValidateTensorIndexing() argument
72 const int tensor_index = tensor_indices[index]; in ValidateTensorIndexing()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_gather_op_test.py | 174 tensor_indices = [0, 1, 2]
182 self.evaluate(ragged_gather_ops.gather(ragged_params, tensor_indices))
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | calibrator.cc | 251 const std::vector<int>& tensor_indices, in GetLoggableTensorIndices() argument
255 for (auto tensor_index : tensor_indices) { in GetLoggableTensorIndices()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 1236 tensor_indices = []
1240 tensor_indices.append(i)
1244 for i, n in zip(tensor_indices, inner_tensor_args):
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder.cc | 2455 const TfLiteIntArray* tensor_indices, in IsAllAllowedTensors() argument
2457 for (int i = 0; i < tensor_indices->size; ++i) { in IsAllAllowedTensors()
2458 int tensor_idx = tensor_indices->data[i]; in IsAllAllowedTensors()
|