Searched refs:GetPrimaryStorageType (Results 1 – 6 of 6) sorted by relevance
38 DeduceLinearStorageType(definition.GetPrimaryStorageType()); in CreatePReLU()52 gpu_info, shape, definition.GetPrimaryStorageType(), in CreatePReLU()
209 SelectBestStorageType(gpu_info, shape, definition.GetPrimaryStorageType(), in CreateElementwiseTwoInput()241 SelectBestStorageType(gpu_info, shape, definition.GetPrimaryStorageType(), in CreateElementwiseTwoInput()
168 DeduceLinearStorageType(definition.GetPrimaryStorageType()); in CreateFullyConnected()
587 DeduceLinearStorageType(definition.GetPrimaryStorageType()); in CreateConvolutionTransposed()604 DeduceLinearStorageType(definition.GetPrimaryStorageType()); in CreateConvolutionTransposed3D()639 desc.storage_type = DeduceLinearStorageType(new_def.GetPrimaryStorageType()); in CreateConvolutionTransposedDynamicWeights()
83 TensorStorageType GetPrimaryStorageType() const;
57 TensorStorageType OperationDef::GetPrimaryStorageType() const { in GetPrimaryStorageType() function in tflite::gpu::OperationDef