HomeSort by relevance Sort by last modified time
    Searched defs:TensorSliceSet (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/util/
tensor_slice_set.cc 28 TensorSliceSet::TensorSliceSet(const TensorShape& shape, DataType type)
31 TensorSliceSet::~TensorSliceSet() {}
33 Status TensorSliceSet::Register(const TensorSlice& slice, const string& tag,
56 TensorSliceSet::SliceInfo info = {slice, tag, data,
63 bool TensorSliceSet::Query(const TensorSlice& slice, float* data) const {
67 const TensorSliceSet::SliceInfo* info = gtl::FindOrNull(slices_, str);
119 bool TensorSliceSet::QueryMeta(
126 const TensorSliceSet::SliceInfo* info = gtl::FindOrNull(slices_, str)
    [all...]
tensor_slice_set.h 40 class TensorSliceSet {
42 TensorSliceSet(const TensorShape& shape, DataType type);
43 virtual ~TensorSliceSet();
95 // Registers "slice" in the TensorSliceSet stored in "tensor_slices", under key
102 std::unordered_map<string, TensorSliceSet*>* tensor_slices);

Completed in 89 milliseconds