OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:ShapeRefiner
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/common_runtime/
shape_refiner.h
84
//
ShapeRefiner
performs shape inference for TensorFlow Graphs. It is
89
class
ShapeRefiner
{
91
ShapeRefiner
(int graph_def_version, const OpRegistryInterface* ops);
93
// Same as
ShapeRefiner
(versions.producer(), ops)
94
ShapeRefiner
(const VersionDef& versions, const OpRegistryInterface* ops);
96
~
ShapeRefiner
();
101
// All inputs of 'node' must be added to
ShapeRefiner
prior to
301
TF_DISALLOW_COPY_AND_ASSIGN(
ShapeRefiner
);
shape_refiner.cc
41
ShapeRefiner
::
ShapeRefiner
(int graph_def_version,
47
ShapeRefiner
::
ShapeRefiner
(const VersionDef& versions,
49
:
ShapeRefiner
(versions.producer(), ops) {}
51
ShapeRefiner
::~
ShapeRefiner
() {
62
// Runs shape inference for the given node using the given
ShapeRefiner
.
65
Status InferShapesForFunctionSubNode(const Node* node,
ShapeRefiner
* refiner,
131
Status
ShapeRefiner
::InferShapesForFunction
[
all
...]
Completed in 200 milliseconds