Searched refs:are_buckets_ready_t (Results 1 – 1 of 1) sorted by relevance
512 Tensor* are_buckets_ready_t = nullptr; in Compute() local514 &are_buckets_ready_t)); in Compute()515 are_buckets_ready_t->scalar<bool>()() = are_buckets_ready; in Compute()580 const Tensor* are_buckets_ready_t = nullptr; in Compute() local582 context->input(kAreBucketsReadyName, &are_buckets_ready_t)); in Compute()583 streams_resource->set_buckets_ready(are_buckets_ready_t->scalar<bool>()()); in Compute()727 Tensor* are_buckets_ready_t = nullptr; in Compute() local730 resource_handle_idx, {}, &are_buckets_ready_t)); in Compute()731 are_buckets_ready_t->scalar<bool>()() = are_buckets_ready; in Compute()