Home
last modified time | relevance | path

Searched refs:tensor_map (Results 1 – 10 of 10) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_scan.cpp89 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()
Dcxx11_tensor_io.cpp112 TensorMap<Tensor<const int, 1, DataLayout> > tensor_map(tensor.data(), 5); in test_output_const() local
115 os << tensor_map; in test_output_const()
Dcxx11_tensor_reduction.cpp334 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/
Dquantize_weights.cc198 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/
Dmodels.py77 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/
Dinterpreter_builder.cc133 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/
Deigen_spatial_convolutions_test.cc1481 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()
DBUILD2785 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/
Dconvert_nodes_test.cc1333 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/
DBUILD722 "//tensorflow/core/kernels:tensor_map",