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()
56 TensorSliceSet::SliceInfo info = {slice, tag, data, in Register()
63 bool TensorSliceSet::Query(const TensorSlice& slice, float* data) const { in Query()
67 const TensorSliceSet::SliceInfo* info = gtl::FindOrNull(slices_, str); in Query()
119 bool TensorSliceSet::QueryMeta( in QueryMeta()
126 const TensorSliceSet::SliceInfo* info = gtl::FindOrNull(slices_, str); in QueryMeta()
174 std::unordered_map<string, TensorSliceSet*>* tensor_slices) { in RegisterTensorSlice()
176 TensorSliceSet* tss = gtl::FindPtrOrNull(*tensor_slices, name); in RegisterTensorSlice()
[all …]
|
| D | tensor_slice_set.h | 40 class TensorSliceSet {
42 TensorSliceSet(const TensorShape& shape, DataType type);
43 virtual ~TensorSliceSet();
102 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_set_test.cc | 44 TensorSliceSet tss(shape, DT_FLOAT); in TEST()
144 TensorSliceSet tss(shape, DT_INT32); in TEST()
242 TensorSliceSet slice_set(shape, DT_INT32); in BM_RegisterOneByOne()
|
| D | tensor_slice_reader.cc | 189 const TensorSliceSet* TensorSliceReader::FindTensorSlice( in FindTensorSlice()
192 const TensorSliceSet* tss = gtl::FindPtrOrNull(tensors_, name); in FindTensorSlice()
204 const TensorSliceSet* tss = gtl::FindPtrOrNull(tensors_, name); in HasTensor()
231 const TensorSliceSet* tss = gtl::FindPtrOrNull(tensors_, name); in GetTensor()
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.h | 297 std::unordered_map<string, checkpoint::TensorSliceSet*> tensor_slices_;
|
| D | tensor_bundle.cc | 935 using checkpoint::TensorSliceSet; in GetSliceValue()
941 const TensorSliceSet* tss = in GetSliceValue()
|