Searched refs:node_ids_t (Results 1 – 2 of 2) sorted by relevance
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()
1253 const Tensor* node_ids_t; in Compute() local1254 OP_REQUIRES_OK(context, context->input("node_ids", &node_ids_t)); in Compute()1255 const auto node_ids = node_ids_t->vec<int32>(); in Compute()1269 const int64 batch_size = node_ids_t->dim_size(0); in Compute()1320 const Tensor* node_ids_t; in Compute() local1321 OP_REQUIRES_OK(context, context->input("node_ids", &node_ids_t)); in Compute()1322 const auto node_ids = node_ids_t->vec<int32>(); in Compute()1340 const int64 batch_size = node_ids_t->dim_size(0); in Compute()1525 const Tensor* node_ids_t; in Compute() local1526 OP_REQUIRES_OK(context, context->input("node_ids", &node_ids_t)); in Compute()[all …]