Searched refs:arg_stypes (Results 1 – 4 of 4) sorted by relevance
428 StorageTypeVector arg_stypes(idx.num_node_entries(), -1); in Init() local438 arg_stypes[eid] = aux_states[aux_top].storage_type(); in Init()446 arg_stypes[eid] = in_args[arg_top].storage_type(); in Init()451 arg_stypes[grad_eid] = arg_grad_store[arg_top].storage_type(); in Init()831 arg_stypes[i] = it3->second; in Init()1653 StorageTypeVector arg_stypes, in CreateCachedSegOpr() argument1816 arg_stypes[i] = it3->second; in CreateCachedSegOpr()1908 StorageTypeVector arg_stypes; // all input stypes in CreateCachedSegOpr() local1909 arg_stypes.reserve(input_names.size()); in CreateCachedSegOpr()1919 arg_stypes.push_back(aux_st.storage_type()); in CreateCachedSegOpr()[all …]
1397 StorageTypeVector arg_stypes(args_len + aux_len); in MXOptimizeForBackend() local1426 arg_stypes[i] = in_arg.storage_type(); in MXOptimizeForBackend()1437 arg_stypes[i] = in_arg.storage_type(); in MXOptimizeForBackend()1451 arg_stypes[i] = it_type->second; in MXOptimizeForBackend()1466 g = exec::InferStorageType(std::move(g), std::move(arg_stypes), "__storage_type__"); in MXOptimizeForBackend()