Searched defs:tensor_dims (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_util.cc | 71 auto tensor_dims = tensor.flat<int64>(); in ShapeTensorToTensorShape() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantization_utils.cc | 164 RuntimeShape tensor_dims{dimension[0], dimension[1], dimension[2], in FillPerChannelMinMax() local
377 RuntimeShape tensor_dims{dimension[0], dimension[1], dimension[2], in SymmetricPerChannelQuantizeValues() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 448 uint32_t* tensor_dims = reinterpret_cast<uint32_t*>(tensor->dims->data); in ConvertTensorTypeToNNType() local
1113 const uint32_t* tensor_dims = reinterpret_cast<const uint32_t*>(dims->data); in AddNewInputConstantTensor() local
1379 uint32_t* tensor_dims = in AddTensor() local
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2393 const int64_t* tensor_dims[] = {&dims1[0], &dims2[0]}; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 4774 std::vector<int> tensor_dims(dims.d, dims.d + dims.nbDims); in ConvertPack() local
|
| D | convert_nodes_test.cc | 2359 std::vector<int> tensor_dims; in TEST_F() member
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1361 int tensor_dims[] = {1, input_size, 1}; in Create() local
|