Searched refs:tensor_ids (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/c/eager/ |
D | tape.h | 141 bool ShouldRecord(gtl::ArraySlice<int64> tensor_ids, 307 bool ShouldRecord(gtl::ArraySlice<int64> tensor_ids, 386 gtl::ArraySlice<int64> tensor_ids, in ShouldRecord() argument 388 CHECK_EQ(tensor_ids.size(), dtypes.size()); in ShouldRecord() 389 for (int i = 0; i < tensor_ids.size(); ++i) { in ShouldRecord() 390 if (tensor_tape_.find(tensor_ids[i]) != tensor_tape_.end()) { in ShouldRecord() 901 gtl::ArraySlice<int64> tensor_ids, in ShouldRecord() argument 906 return call_state_.top().backward_tape->ShouldRecord(tensor_ids, dtypes); in ShouldRecord() 911 for (int i = 0; i < tensor_ids.size(); ++i) { in ShouldRecord() 912 if (accumulated_gradients_.find(tensor_ids[i]) != in ShouldRecord()
|
D | gradients.cc | 250 std::vector<int64> tensor_ids(tensors.size()); in ShouldRecord() local 253 tensor_ids[i] = ToId(tensors[i]); in ShouldRecord() 256 return GradientTape::ShouldRecord(tensor_ids, tensor_dtypes); in ShouldRecord()
|
/external/tensorflow/tensorflow/python/debug/lib/ |
D | dumping_callback.py | 339 tensor_ids): argument 429 tensor_id=tensor_ids[output_slot], 657 tensor_ids = [] 662 tensor_ids.append(self._symbolic_tensor_counter) 663 return tensor_ids
|
/external/tensorflow/tensorflow/python/eager/ |
D | pywrap_tfe_src.cc | 1914 std::vector<tensorflow::int64> tensor_ids; local 1915 tensor_ids.reserve(len); 1924 tensor_ids.push_back(id); 1926 tensor_ids.push_back(-1); 1930 return tensor_ids; 1936 std::vector<tensorflow::int64>* tensor_ids, argument 1945 tensor_ids->reserve(len); 1949 tensor_ids->push_back(FastTensorId(item)); 1978 std::vector<tensorflow::int64> tensor_ids; local 1980 if (!TensorShapesAndDtypes(tensors, &tensor_ids, &dtypes)) { [all …]
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | deadness_analysis.cc | 1552 std::vector<TensorId> tensor_ids; in Print() local 1554 tensor_ids.push_back(kv_pair.first); in Print() 1557 std::sort(tensor_ids.begin(), tensor_ids.end()); in Print() 1559 for (TensorId tensor_id : tensor_ids) { in Print()
|