OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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