Searched defs:tensor_dims (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantization_utils.cc | 91 RuntimeShape tensor_dims{dimension[0], dimension[1], dimension[2], in SymmetricPerChannelQuantization() local
143 RuntimeShape tensor_dims{dimension[0], dimension[1], dimension[2], in SymmetricPerChannelQuantizeValues() local
|
| D | subgraph_quantizer.cc | 111 const std::vector<int> tensor_dims = {tensor->shape[0], tensor->shape[1], in SymmetricPerChannelQuantizeTensor() local
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2346 const int64_t* tensor_dims[] = {&dims1[0], &dims2[0]}; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1526 std::vector<int> tensor_dims; in TEST_F() member
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1162 int tensor_dims[] = {1, input_size, 1}; in Create() local
|