Searched refs:node_ids_t (Results 1 – 2 of 2) sorted by relevance
219 const Tensor* node_ids_t; in Compute() local220 OP_REQUIRES_OK(context, context->input("node_ids", &node_ids_t)); in Compute()221 const auto node_ids = node_ids_t->vec<int32>(); in Compute()235 const int64 batch_size = node_ids_t->dim_size(0); in Compute()
153 using node_ids_t = std::vector<node_ids_elem_t, ArenaAllocator<node_ids_elem_t>>; in SemistaticGraph() local154 node_ids_t node_ids = node_ids_t(ArenaAllocator<node_ids_elem_t>(memory_pool)); in SemistaticGraph()