Searched refs:tensor_map (Results 1 – 7 of 7) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_scan.cpp | 89 TensorMap<Tensor<int, 1, DataLayout> > tensor_map(inputs, 20); in test_tensor_maps() local
90 tensor_map.setRandom(); in test_tensor_maps()
92 Tensor<int, 1, DataLayout> result = tensor_map.cumsum(0); in test_tensor_maps()
96 accum += tensor_map(i); in test_tensor_maps()
|
| D | cxx11_tensor_io.cpp | 112 TensorMap<Tensor<const int, 1, DataLayout> > tensor_map(tensor.data(), 5); in test_output_const() local
115 os << tensor_map; in test_output_const()
|
| D | cxx11_tensor_reduction.cpp | 334 TensorMap<Tensor<int, 4, DataLayout> > tensor_map(inputs, 2, 3, 5, 7); in test_tensor_maps() local
340 tensor_map.setRandom(); in test_tensor_maps()
345 Tensor<int, 2, DataLayout> result = tensor_map.sum(reduction_axis); in test_tensor_maps()
355 sum += tensor_map(i, k, j, l); in test_tensor_maps()
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | models.py | 88 tensor_map = {} # Map {reference_tensor: corresponding_tensor}
124 tensor_map[x] = y
152 tensor in tensor_map for tensor in nest.flatten(node.input_tensors)):
153 computed_tensors = nest.map_structure(lambda t: tensor_map[t],
161 tensor_map[x] = y
167 assert x in tensor_map, 'Could not compute output ' + str(x)
168 output_tensors.append(tensor_map[x])
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_weights.cc | 150 std::unordered_map<int32_t, TensorT*>* tensor_map) { in InsertQuantizableInputTensorsFromOperator() argument
188 tensor_map->insert({tensor_idx, tensor}); in InsertQuantizableInputTensorsFromOperator()
270 std::unordered_map<int32_t, TensorT*> tensor_map; in QuantizeWeightsInternal() local
274 model.get(), op, weights_min_num_elements, &tensor_map)); in QuantizeWeightsInternal()
280 for (std::pair<int32_t, TensorT*> tensor_pair : tensor_map) { in QuantizeWeightsInternal()
287 for (const auto& tensor_pair : tensor_map) { in QuantizeWeightsInternal()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_spatial_convolutions_test.cc | 1475 ArgType tensor_map(inputs[i].data(), input_dims); in PackRhsHelper() local
1479 tensor_map, // in PackRhsHelper()
1659 ArgType tensor_map(filters[i].data(), filter_dims); in PackLhsHelper() local
1662 TensorReshapingOp<NewDimension, ArgType>(tensor_map, reshape_dims); in PackLhsHelper()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1061 const auto& tensor_map = data.tensor.flat<T>(); in GetSpanForData() local
1062 return absl::Span<const T>(tensor_map.data(), tensor_map.size()); in GetSpanForData()
|