Searched refs:unowned_indices (Results 1 – 3 of 3) sorted by relevance
113 bool aliased = arg.unowned_indices().count(pair.first) > 0; in ExecuteAsyncOnStream()123 const auto& unowned_indices = arg.unowned_indices(); in ExecuteAsyncOnStream() local124 se_args[i]->unowned_indices_size = unowned_indices.size(); in ExecuteAsyncOnStream()125 se_args[i]->unowned_indices = new XLA_ShapeIndex[unowned_indices.size()]; in ExecuteAsyncOnStream()127 for (auto& idx : unowned_indices) { in ExecuteAsyncOnStream()128 se_args[i]->unowned_indices[j] = ApiConverter::ToC(idx); in ExecuteAsyncOnStream()145 delete[] se_args[i]->unowned_indices; in ExecuteAsyncOnStream()
257 XLA_ShapeIndex* unowned_indices; member
117 const std::set<ShapeIndex>& unowned_indices() { return unowned_indices_; } in unowned_indices() function