Searched refs:num_streams_t (Results 1 – 1 of 1) sorted by relevance
116 const Tensor* num_streams_t; in Compute() local117 OP_REQUIRES_OK(context, context->input(kNumStreamsName, &num_streams_t)); in Compute()118 int64_t num_streams = num_streams_t->scalar<int64>()(); in Compute()