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

  /external/tensorflow/tensorflow/python/training/saving/
functional_saver.py 88 tensor_slices = []
93 tensor_slices.append(spec.slice_spec)
96 file_prefix, tensor_names, tensor_slices, tensors)]):
127 tensor_names, tensor_slices, tensor_dtypes = zip(*restore_specs)
130 file_prefix, tensor_names, tensor_slices, tensor_dtypes)
  /external/tensorflow/tensorflow/core/util/
tensor_slice_set.cc 174 std::unordered_map<string, TensorSliceSet*>* tensor_slices) {
175 DCHECK_NE(tensor_slices, nullptr);
176 TensorSliceSet* tss = gtl::FindPtrOrNull(*tensor_slices, name);
180 tensor_slices->insert(std::make_pair(name, tss));
tensor_slice_set.h 95 // Registers "slice" in the TensorSliceSet stored in "tensor_slices", under key
98 // REQUIRES: tensor_slices != nullptr
102 std::unordered_map<string, TensorSliceSet*>* tensor_slices);
  /external/tensorflow/tensorflow/core/kernels/
load_and_remap_matrix_op.cc 169 std::vector<TensorSlice> tensor_slices; variable
185 tensor_slices.push_back(slice);
203 for (const TensorSlice& tensor_slice : tensor_slices) {
  /external/tensorflow/tensorflow/python/ops/
io_ops.py 42 def _save(filename, tensor_names, tensors, tensor_slices=None, name="save"):
50 tensor_slices=["4 10 0,2:-", "4 10 2,2:-"])
56 tensor_slices: Optional list of strings to specify the shape and slices of
63 tensor_slices.
68 if tensor_slices is None:
71 return gen_io_ops.save_slices(filename, tensor_names, tensor_slices,
  /external/tensorflow/tensorflow/python/training/
saver.py 107 tensor_slices = []
112 tensor_slices.append(spec.slice_spec)
118 tensor_slices=tensor_slices)
122 return io_ops.save_v2(filename_tensor, tensor_names, tensor_slices,
    [all...]

Completed in 783 milliseconds