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()
|