Home
last modified time | relevance | path

Searched refs:unused_tensors (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/eager/
Dgradient_input_output_exclusions.py328 unused_tensors = all_tensors - used_tensors
329 if unused_tensors:
330 unused_tensor_list = sorted(list(unused_tensors))
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib.h108 void SetUnusedTensors(const std::set<int>& unused_tensors) { in SetUnusedTensors() argument
109 unused_tensors_ = unused_tensors; in SetUnusedTensors()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dsnapshot_util.cc349 std::vector<Tensor> unused_tensors; in SkipRecords() local
350 TF_RETURN_IF_ERROR(ReadTensors(&unused_tensors)); in SkipRecords()