Searched defs:out_dtype (Results 1 – 5 of 5) sorted by relevance
107 dtype out_dtype = dtype::get_builtin<TResult>(); in call() local170 dtype out_dtype = dtype::get_builtin<TResult>(); in call() local
35 auto out_dtype = kFloat32; in FftRealInfer() local
156 def infer_dtype(self, x_dtype, out_dtype, dout_dtype): argument
128 def infer_dtype(self, x_dtype, out_dtype, dout_dtype): argument
551 auto out_dtype = output[ATTR_DTYPE]; in PyInferRes2Abstract() local