Searched refs:tensor_ids (Results 1 – 3 of 3) sorted by relevance
124 bool ShouldRecord(gtl::ArraySlice<int64> tensor_ids,186 gtl::ArraySlice<int64> tensor_ids, in ShouldRecord() argument188 CHECK_EQ(tensor_ids.size(), dtypes.size()); in ShouldRecord()189 for (int i = 0; i < tensor_ids.size(); ++i) { in ShouldRecord()190 if (tensor_tape_.find(tensor_ids[i]) != tensor_tape_.end()) { in ShouldRecord()
1155 std::vector<TensorId> tensor_ids; in Print() local1157 tensor_ids.push_back(kv_pair.first); in Print()1160 std::sort(tensor_ids.begin(), tensor_ids.end()); in Print()1162 for (TensorId tensor_id : tensor_ids) { in Print()1190 std::vector<TensorId> tensor_ids; in PredicateMapAsString() local
1381 std::vector<tensorflow::int64> tensor_ids; local1382 tensor_ids.reserve(len);1391 tensor_ids.push_back(id);1393 tensor_ids.push_back(-1);1397 return tensor_ids;1418 std::vector<tensorflow::int64> tensor_ids; local1420 tensor_ids.reserve(len);1424 tensor_ids.push_back(FastTensorId(item));1430 if (tape->tape->ShouldRecord(tensor_ids, dtypes)) {