| /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 | 336 TensorMap<Tensor<int, 4, DataLayout> > tensor_map(inputs, 2, 3, 5, 7); in test_tensor_maps() local
342 tensor_map.setRandom(); in test_tensor_maps()
347 Tensor<int, 2, DataLayout> result = tensor_map.sum(reduction_axis); in test_tensor_maps()
357 sum += tensor_map(i, k, j, l); in test_tensor_maps()
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | models.py | 73 def _make_new_nodes(nodes_by_depth, layer_fn, layer_map, tensor_map): argument
110 tensor in tensor_map for tensor in nest.flatten(node.input_tensors)):
112 args = nest.map_structure(lambda t: tensor_map.get(t, t),
114 kwargs = nest.map_structure(lambda t: tensor_map.get(t, t),
125 tensor_map[x] = y
361 tensor_map = {} # Maps tensors from `model` to those in `cloned_model`.
369 tensor_map[node.output_tensors[j]] = output_tensor
371 tensor_map[node.output_tensors] = cloned_node.output_tensors
378 }, layer_fn, layer_map, tensor_map)
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_weights.cc | 199 absl::flat_hash_map<int32_t, TensorPerChannel>* tensor_map, in InsertQuantizableInputTensorsFromOperator() argument
240 tensor_map->insert({tensor_idx, in InsertQuantizableInputTensorsFromOperator()
244 tensor_map->insert({tensor_idx, in InsertQuantizableInputTensorsFromOperator()
288 tensor_map->insert({tensor_idx, {tensor, /*is_per_channel=*/false}}); in InsertQuantizableInputTensorsFromOperator()
439 absl::flat_hash_map<int32_t, TensorPerChannel> tensor_map; local
443 model.get(), op, weights_min_num_elements, custom_op_map, &tensor_map,
447 for (std::pair<int32_t, TensorPerChannel> tensor_pair : tensor_map) {
460 for (const auto& tensor_pair : tensor_map) {
564 absl::flat_hash_map<int32_t, TensorT*> tensor_map; in QuantizeWeightsFloat16() local
580 tensor_map.insert({tensor_idx, tensor}); in QuantizeWeightsFloat16()
[all …]
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_builder.cc | 135 tensor_map) { in GetMapFromTensorMap()
136 if (!tensor_map) return {}; in GetMapFromTensorMap()
138 for (const auto tensor : *tensor_map) { in GetMapFromTensorMap()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_spatial_convolutions_test.cc | 1481 ArgType tensor_map(inputs[i].data(), input_dims); in PackRhsHelper() local
1485 tensor_map, // in PackRhsHelper()
1686 ArgType tensor_map(filters[i].data(), filter_dims); in PackLhsHelper() local
1689 TensorReshapingOp<NewDimension, ArgType>(tensor_map, reshape_dims); in PackLhsHelper()
|
| D | BUILD | 2857 name = "tensor_map",
2858 srcs = ["tensor_map.cc"],
2859 hdrs = ["tensor_map.h"],
2877 ":tensor_map",
2893 ":tensor_map",
6134 "tensor_map.h",
6457 "tensor_map.cc",
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1222 const auto& tensor_map = data.tensor.flat<T>(); in GetSpanForData() local
1223 return absl::Span<const T>(tensor_map.data(), tensor_map.size()); in GetSpanForData()
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 731 "//tensorflow/core/kernels:tensor_map",
|