HomeSort by relevance Sort by last modified time
    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