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