Searched refs:tensor_slices (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | 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/ |
| D | tensor_slice_set.cc | 174 std::unordered_map<string, TensorSliceSet*>* tensor_slices) { in RegisterTensorSlice() argument
175 DCHECK_NE(tensor_slices, nullptr); in RegisterTensorSlice()
176 TensorSliceSet* tss = gtl::FindPtrOrNull(*tensor_slices, name); in RegisterTensorSlice()
180 tensor_slices->insert(std::make_pair(name, tss)); in RegisterTensorSlice()
|
| D | tensor_slice_set.h | 102 std::unordered_map<string, TensorSliceSet*>* tensor_slices);
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | io_ops.py | 42 def _save(filename, tensor_names, tensors, tensor_slices=None, name="save"): argument
68 if tensor_slices is None:
71 return gen_io_ops.save_slices(filename, tensor_names, tensor_slices,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | load_and_remap_matrix_op.cc | 169 std::vector<TensorSlice> tensor_slices; in Compute() local
185 tensor_slices.push_back(slice); in Compute()
203 for (const TensorSlice& tensor_slice : tensor_slices) { in Compute()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | 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,
|