Searched refs:fetch_and_cast (Results 1 – 4 of 4) sorted by relevance
69 C10_HOST_DEVICE inline dest_t fetch_and_cast( in fetch_and_cast() function106 C10_HOST_DEVICE inline T fetch_and_cast<T>( \
386 return c10::fetch_and_cast<T>(op.tensor_base().scalar_type(), op.data); in scalar_value()399 return c10::fetch_and_cast<T>( in original_scalar_value()
232 return f(c10::fetch_and_cast<typename traits::template arg<I>::type>( in invoke_impl()
123 return c10::fetch_and_cast<scalar_t>(dtypes[arg], ptr); in load()