Searched defs:all_tensors (Results 1 – 2 of 2) sorted by relevance
54 def print_tensors_in_checkpoint_file(file_name, tensor_name, all_tensors, argument
280 std::vector<TensorId> all_tensors() const in all_tensors() function