Searched defs:device_dtype (Results 1 – 1 of 1) sorted by relevance
236 …Dtype(const std::map<TypeId, TypeId> &type_map, const TypeId &in_dtype, const TypeId &device_dtype, in CheckHitTargetDtype()262 auto device_dtype = kernel_build_info->GetInputDeviceType(input_index); in TagRaiseReduce() local273 auto device_dtype = kernel_build_info->GetOutputDeviceType(output_index); in TagRaiseReduce() local