HomeSort by relevance Sort by last modified time
    Searched refs:all_tensors (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/python/tools/
inspect_checkpoint.py 32 def print_tensors_in_checkpoint_file(file_name, tensor_name, all_tensors,
44 all_tensors: Boolean indicating whether to print all tensors.
49 if all_tensors or all_tensor_names:
53 if all_tensors:
112 "[--all_tensors] "
118 FLAGS.all_tensors, FLAGS.all_tensor_names)
137 "--all_tensors",
  /external/tensorflow/tensorflow/python/ops/
custom_gradient.py 213 all_tensors = flat_result + args + variables
238 original_tensors = all_tensors
240 all_tensors = array_ops.identity_n(all_tensors)
247 all_tensors[i]._handle_data = t._handle_data # pylint: disable=protected-access
249 f.__name__, all_tensors, original_tensors, tape_grad_fn)
250 for ot, t in zip(original_tensors, all_tensors):
253 structure=result, flat_sequence=all_tensors[:len(flat_result)])
  /external/tensorflow/tensorflow/contrib/graph_editor/
subgraph.py 203 all_tensors = frozenset(inputs + outputs + list(insides))
204 self._passthrough_ts = [t for t in passthrough_ts if t not in all_tensors]
  /external/tensorflow/tensorflow/python/training/
saver.py 146 all_tensors = []
153 all_tensors.extend(
155 return all_tensors
330 all_tensors = self.bulk_restore(filename_tensor, saveables, preferred_shard,
351 saveable_tensors = all_tensors[idx:idx + len(saveable.specs)]
    [all...]
  /external/tensorflow/tensorflow/lite/testing/
generate_examples.py     [all...]
  /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
convert_nodes.cc 1322 std::set<nvinfer1::ITensor*> all_tensors; local
    [all...]

Completed in 1054 milliseconds