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

  /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 87 def _save(filename, tensor_names, tensors, tensor_slices=None, name="save"):
95 tensor_slices=["4 10 0,2:-", "4 10 2,2:-"])
101 tensor_slices: Optional list of strings to specify the shape and slices of
108 tensor_slices.
113 if tensor_slices is None:
116 return gen_io_ops._save_slices(filename, tensor_names, tensor_slices,
  /external/tensorflow/tensorflow/python/training/
saver.py 225 tensor_slices = []
230 tensor_slices.append(spec.slice_spec)
236 tensor_slices=tensor_slices)
240 return io_ops.save_v2(filename_tensor, tensor_names, tensor_slices,
    [all...]

Completed in 101 milliseconds