Searched refs:SelectBestStorageType (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | storage_type_util.h | 34 TensorStorageType SelectBestStorageType(const GpuInfo& gpu_info,
|
| D | storage_type_util.cc | 78 TensorStorageType SelectBestStorageType(const GpuInfo& gpu_info, in SelectBestStorageType() function
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | elementwise.cc | 209 SelectBestStorageType(gpu_info, shape, definition.GetPrimaryStorageType(), in CreateElementwiseTwoInput()
241 SelectBestStorageType(gpu_info, shape, definition.GetPrimaryStorageType(), in CreateElementwiseTwoInput()
|
| D | prelu.cc | 51 TensorStorageType storage_type = SelectBestStorageType( in CreatePReLU()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | operation_selector.cc | 98 td_0.storage_type = SelectBestStorageType( in WinogradFromNode()
104 td_1.storage_type = SelectBestStorageType( in WinogradFromNode()
213 transposed_desc.storage_type = SelectBestStorageType( in GPUOperationFromNode()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | inference_context.cc | 190 SelectBestStorageType(gpu_info, shape, storage_type, data_type, layout); in ReserveGraphTensors()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | inference_context.cc | 283 SelectBestStorageType(gpu_info, shape, storage_type, data_type, layout); in ReserveGraphTensors()
|