Searched refs:in_aux_ptr (Results 1 – 8 of 8) sorted by relevance
199 in_aux_ptr = g.GetAttr<NDArray**>("in_aux"); in PrePartition()245 if (in_aux_ptr[i] != nullptr) { in PrePartition()247 const auto &in_aux = *(in_aux_ptr[i]); in PrePartition()540 NDArray **in_aux_ptr; variable
348 NDArray **in_aux_ptr = g.GetAttr<NDArray**>("in_aux"); in PrePartition() local356 in_aux_dict_[in_aux_names[i]] = in_aux_ptr[i]; in PrePartition()
1383 NDArray **in_aux_ptr = reinterpret_cast<NDArray**>(in_aux_handle); in MXOptimizeForBackend() local1422 if (in_aux_ptr[aux_top] != nullptr) { in MXOptimizeForBackend()1423 const auto &in_arg = *(in_aux_ptr[aux_top]); in MXOptimizeForBackend()1474 g.attrs["in_aux"] = std::make_shared<nnvm::any>(in_aux_ptr); in MXOptimizeForBackend()1483 NDArray **in_aux_ptr = static_cast<NDArray**>(nullptr); in MXOptimizeForBackend() local1485 g.attrs["in_aux"] = std::make_shared<nnvm::any>(in_aux_ptr); in MXOptimizeForBackend()
1329 NDArray **in_aux_ptr = g.GetAttr<NDArray**>("in_aux"); in registerPasses() local1413 if (in_aux_ptr[i] != nullptr) { in registerPasses()1415 const auto &in_aux = *(in_aux_ptr[i]); in registerPasses()