Searched refs: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() argument240 if (iter == type_map.end() && in_dtype != device_dtype) { in CheckHitTargetDtype()245 if (iter != type_map.end() && iter->second != device_dtype && in_dtype != device_dtype) { in CheckHitTargetDtype()248 if (in_dtype == kNumberTypeInt64 && device_dtype == kNumberTypeInt32) { in CheckHitTargetDtype()262 auto device_dtype = kernel_build_info->GetInputDeviceType(input_index); in TagRaiseReduce() local263 if (device_dtype == kNumberTypeFloat || device_dtype == kNumberTypeFloat32) { in TagRaiseReduce()264 device_dtype = kNumberTypeFloat32; in TagRaiseReduce()266 if (!CheckHitTargetDtype(type_map, in_dtype, device_dtype, &flag)) { in TagRaiseReduce()273 auto device_dtype = kernel_build_info->GetOutputDeviceType(output_index); in TagRaiseReduce() local274 if (device_dtype == kNumberTypeFloat || device_dtype == kNumberTypeFloat32) { in TagRaiseReduce()[all …]