Searched defs:cast_dtype (Results 1 – 2 of 2) sorted by relevance
38 inline aotriton::DType cast_dtype(caffe2::TypeMeta t_dtype) in cast_dtype() function
200 const DataType cast_dtype = opts.should_cast ? opts.should_cast(edge) : dtype; in AddSend() local260 DataType cast_dtype = dtype; in AddRecv() local