| /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/lite/tools/optimize/ |
| D | quantize_weights.cc | 198 absl::flat_hash_map<int32_t, TensorPerChannel>* tensor_map, in InsertQuantizableInputTensorsFromOperator() argument
239 tensor_map->insert({tensor_idx, in InsertQuantizableInputTensorsFromOperator()
243 tensor_map->insert({tensor_idx, in InsertQuantizableInputTensorsFromOperator()
287 tensor_map->insert({tensor_idx, {tensor, /*is_per_channel=*/false}}); in InsertQuantizableInputTensorsFromOperator()
434 absl::flat_hash_map<int32_t, TensorPerChannel> tensor_map; in QuantizeWeightsInt8() local
438 model.get(), op, weights_min_num_elements, custom_op_map, &tensor_map, in QuantizeWeightsInt8()
442 for (std::pair<int32_t, TensorPerChannel> tensor_pair : tensor_map) { in QuantizeWeightsInt8()
455 for (const auto& tensor_pair : tensor_map) { in QuantizeWeightsInt8()
558 absl::flat_hash_map<int32_t, TensorT*> tensor_map; in QuantizeWeightsFloat16() local
574 tensor_map.insert({tensor_idx, tensor}); in QuantizeWeightsFloat16()
[all …]
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | models.py | 77 def _make_new_nodes(nodes_by_depth, layer_fn, layer_map, tensor_map): argument
114 tensor in tensor_map for tensor in nest.flatten(node.input_tensors)):
116 args = nest.map_structure(lambda t: tensor_map.get(t, t),
118 kwargs = nest.map_structure(lambda t: tensor_map.get(t, t),
129 tensor_map[x] = y
365 tensor_map = {} # Maps tensors from `model` to those in `cloned_model`.
373 tensor_map[node.output_tensors[j]] = output_tensor
375 tensor_map[node.output_tensors] = cloned_node.output_tensors
382 }, layer_fn, layer_map, tensor_map)
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_builder.cc | 133 tensor_map) { in GetMapFromTensorMap()
134 if (!tensor_map) return {}; in GetMapFromTensorMap()
136 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 | 2785 name = "tensor_map",
2786 srcs = ["tensor_map.cc"],
2787 hdrs = ["tensor_map.h"],
2805 ":tensor_map",
2821 ":tensor_map",
5959 "tensor_map.h",
6244 "tensor_map.cc",
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1333 const auto& tensor_map = data.tensor.flat<T>(); in GetSpanForData() local
1334 return absl::Span<const T>(tensor_map.data(), tensor_map.size()); in GetSpanForData()
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 722 "//tensorflow/core/kernels:tensor_map",
|