Searched refs:tensor_indices (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/profiling/ |
| D | profile_summarizer.cc | 41 const TfLiteIntArray* tensor_indices) { in GetTensorNames() argument
43 tensors.reserve(tensor_indices->size); in GetTensorNames()
44 for (int i = 0; i < tensor_indices->size; i++) { in GetTensorNames()
45 tensors.push_back(GetTensorName(interpreter, tensor_indices->data[i])); in GetTensorNames()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_array_ops.cc | 627 const Tensor* tensor_indices; in Compute() local
628 OP_REQUIRES_OK(ctx, ctx->input("indices", &tensor_indices)); in Compute()
629 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(tensor_indices->shape()), in Compute()
632 tensor_indices->shape().DebugString())); in Compute()
633 const auto indices_t = tensor_indices->vec<int32>(); in Compute()
634 num_indices = tensor_indices->NumElements(); in Compute()
1075 const Tensor* tensor_indices; in Compute() local
1076 OP_REQUIRES_OK(ctx, ctx->input("indices", &tensor_indices)); in Compute()
1077 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(tensor_indices->shape()), in Compute()
1080 tensor_indices->shape().DebugString())); in Compute()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_gather_op_test.py | 118 tensor_indices = [0, 1, 2]
126 self.evaluate(ragged_gather_ops.gather(ragged_params, tensor_indices))
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer.py | 1631 tensor_indices = nest.map_structure(lambda t: t._keras_history[2],
1639 tensor_indices=tensor_indices,
1899 tensor_indices, argument
1924 self.tensor_indices = tensor_indices
1962 nest.flatten(self.tensor_indices), nest.flatten(self.input_tensors))
1971 'tensor_indices': self.tensor_indices
|
| D | input_layer.py | 131 tensor_indices=[],
|
| D | network.py | 300 tensor_indices=[],
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | calibrator.cc | 195 const std::vector<int>& tensor_indices, in GetLoggableTensorIndices() argument
199 for (auto tensor_index : tensor_indices) { in GetLoggableTensorIndices()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 1016 tensor_indices = []
1020 tensor_indices.append(i)
1024 for i, n in zip(tensor_indices, inner_tensor_args):
|