Searched defs:tensor_map (Results 1 – 8 of 8) 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
|
| D | cxx11_tensor_io.cpp | 112 TensorMap<Tensor<const int, 1, DataLayout> > tensor_map(tensor.data(), 5); in test_output_const() local
|
| D | cxx11_tensor_reduction.cpp | 334 TensorMap<Tensor<int, 4, DataLayout> > tensor_map(inputs, 2, 3, 5, 7); in test_tensor_maps() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_weights.cc | 198 absl::flat_hash_map<int32_t, TensorPerChannel>* tensor_map, in InsertQuantizableInputTensorsFromOperator()
434 absl::flat_hash_map<int32_t, TensorPerChannel> tensor_map; in QuantizeWeightsInt8() local
558 absl::flat_hash_map<int32_t, TensorT*> tensor_map; in QuantizeWeightsFloat16() local
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | models.py | 77 def _make_new_nodes(nodes_by_depth, layer_fn, layer_map, tensor_map): argument
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_builder.cc | 132 const flatbuffers::Vector<flatbuffers::Offset<tflite::TensorMap>>* 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
1686 ArgType tensor_map(filters[i].data(), filter_dims); in PackLhsHelper() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1333 const auto& tensor_map = data.tensor.flat<T>(); in GetSpanForData() local
|