Searched refs:isvalid_host_tensor (Results 1 – 1 of 1) sorted by relevance
825 Tensor isvalid_host_tensor; in RunIfBoxIndexIsValid() local834 &isvalid_host_tensor, alloc_attr), in RunIfBoxIndexIsValid()840 isvalid_host_tensor.scalar<bool>().data() /* destination */, in RunIfBoxIndexIsValid()851 auto wrapped_callback = [context, isvalid_host_tensor, isvalid_dev_ref, in RunIfBoxIndexIsValid()855 const bool isvalid = isvalid_host_tensor.scalar<bool>()(); in RunIfBoxIndexIsValid()