Searched defs:tensor_map (Results 1 – 6 of 6) 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 | 150 std::unordered_map<int32_t, TensorT*>* tensor_map) { in InsertQuantizableInputTensorsFromOperator()
270 std::unordered_map<int32_t, TensorT*> tensor_map; in QuantizeWeightsInternal() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_spatial_convolutions_test.cc | 1475 ArgType tensor_map(inputs[i].data(), input_dims); in PackRhsHelper() local
1659 ArgType tensor_map(filters[i].data(), filter_dims); in PackLhsHelper() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1061 const auto& tensor_map = data.tensor.flat<T>(); in GetSpanForData() local
|