Searched refs:isvalid_host_tensor (Results 1 – 1 of 1) sorted by relevance
852 Tensor isvalid_host_tensor; in RunIfBoxIndexIsValid() local861 &isvalid_host_tensor, alloc_attr), in RunIfBoxIndexIsValid()867 isvalid_host_tensor.scalar<bool>().data() /* destination */, in RunIfBoxIndexIsValid()878 auto wrapped_callback = [context, isvalid_host_tensor, isvalid_dev_ref, in RunIfBoxIndexIsValid()882 const bool isvalid = isvalid_host_tensor.scalar<bool>()(); in RunIfBoxIndexIsValid()