Searched refs:SetOutputDeviceType (Results 1 – 3 of 3) sorted by relevance
185 void SetOutputDeviceType(const TypeId &output_device_type, size_t index);
255 void KernelBuildInfo::KernelBuildInfoBuilder::SetOutputDeviceType(const TypeId &output_device_type,… in SetOutputDeviceType() function in mindspore::kernel::KernelBuildInfo::KernelBuildInfoBuilder
103 builder->SetOutputDeviceType(kNumberTypeInt32, output_index); in ReducePrecision()