Searched defs:atten_mask (Results 1 – 4 of 4) sorted by relevance
158 TensorPtr& atten_mask, in run_model_step()278 auto atten_mask = from_blob( in generate() local
128 InfoAttrs atten_mask; member
105 Result<TensorInfo> atten_mask = methods_meta[0]->input_tensor_meta(1); in prepare_io() local267 Result<TensorInfo> atten_mask = methods_meta[0]->input_tensor_meta(1); in prepare_io() local
97 Result<TensorInfo> atten_mask = methods_meta[0]->input_tensor_meta(2); in prepare_io() local