Searched refs:TensorSliceSet (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_set.cc | 28 TensorSliceSet::TensorSliceSet(const TensorShape& shape, DataType type) in TensorSliceSet() function in tensorflow::checkpoint::TensorSliceSet
31 TensorSliceSet::~TensorSliceSet() {} in ~TensorSliceSet()
33 Status TensorSliceSet::Register(const TensorSlice& slice, const string& tag) { in Register()
55 TensorSliceSet::SliceInfo info = {slice, tag, result_shape.num_elements()}; in Register()
60 bool TensorSliceSet::QueryMeta( in QueryMeta()
67 const TensorSliceSet::SliceInfo* info = gtl::FindOrNull(slices_, str); in QueryMeta()
115 std::unordered_map<string, TensorSliceSet*>* tensor_slices) { in RegisterTensorSlice()
117 TensorSliceSet* tss = gtl::FindPtrOrNull(*tensor_slices, name); in RegisterTensorSlice()
120 tss = new TensorSliceSet(shape, type); in RegisterTensorSlice()
|
| D | tensor_slice_set.h | 37 class TensorSliceSet {
39 TensorSliceSet(const TensorShape& shape, DataType type);
40 virtual ~TensorSliceSet();
87 std::unordered_map<string, TensorSliceSet*>* tensor_slices);
|
| D | tensor_slice_reader.h | 95 const std::unordered_map<string, TensorSliceSet*>& Tensors() const { in Tensors()
122 const TensorSliceSet* FindTensorSlice(
135 mutable std::unordered_map<string, TensorSliceSet*> tensors_;
148 const TensorSliceSet* tss; in CopySliceData()
|
| D | tensor_slice_reader.cc | 189 const TensorSliceSet* TensorSliceReader::FindTensorSlice( in FindTensorSlice()
192 const TensorSliceSet* tss = gtl::FindPtrOrNull(tensors_, name); in FindTensorSlice()
209 const TensorSliceSet* tss = gtl::FindPtrOrNull(tensors_, name); in HasTensor()
236 const TensorSliceSet* tss = gtl::FindPtrOrNull(tensors_, name); in GetTensor()
|
| D | tensor_slice_set_test.cc | 43 TensorSliceSet tss(shape, DT_INT32); in TEST()
149 TensorSliceSet slice_set(shape, DT_INT32); in BM_RegisterOneByOne()
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.h | 299 std::unordered_map<string, checkpoint::TensorSliceSet*> tensor_slices_;
|
| D | tensor_bundle.cc | 995 using checkpoint::TensorSliceSet; in GetSliceValue()
1001 const TensorSliceSet* tss = in GetSliceValue()
|