| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.h | 112 Graph* graph, ShapeRefiner* shape_refiner);
116 const ShapeRefiner& shape_refiner, const Node& node,
120 void RegisterConstantNode(const ShapeRefiner& shape_refiner,
140 const ShapeRefiner& shape_refiner);
144 const ShapeRefiner& shape_refiner, const Node& node);
148 const ShapeRefiner& shape_refiner, const Node& node);
151 const ShapeRefiner& shape_refiner, const Node& node);
154 const ShapeRefiner& shape_refiner, const Node& node);
158 const ShapeRefiner& shape_refiner, const Node& node);
162 const ShapeRefiner& shape_refiner, const Node& node);
[all …]
|
| D | graph_transferer.cc | 90 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in LoadGraphFromProto()
317 Graph* graph, ShapeRefiner* shape_refiner) { in TransformGraphToAddAggregatedInputNode()
416 const ShapeRefiner& shape_refiner, const Node& node, in RegisterNode()
449 void GraphTransferer::RegisterConstantNode(const ShapeRefiner& shape_refiner, in RegisterConstantNode()
598 const ShapeRefiner& shape_refiner) { in IsNodeFlattenReshape()
638 const ShapeRefiner& shape_refiner, const Node& node) { in RegisterNodeWithPaddingAndStrides()
672 const ShapeRefiner& shape_refiner, const Node& node) { in RegisterNodeWithRank()
705 const ShapeRefiner& shape_refiner, const Node& node) { in RegisterPadNode()
780 const ShapeRefiner& shape_refiner, const Node& node) { in RegisterInputNode()
798 const ShapeRefiner& shape_refiner, const Node& node) { in RegisterFlattenNode()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner_test.cc | 51 return ShapeRefiner::SameDefinedShape(c, s0, s1); in SameDefinedShape()
58 return ShapeRefiner::IsUpdatedShapesOrTypes(c, existing, updated); in IsUpdatedShapesOrTypes()
61 static constexpr int64 kMaxTensorSize = ShapeRefiner::kMaxTensorSize;
95 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
102 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
119 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
134 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
151 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
196 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
214 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
[all …]
|
| D | shape_refiner.cc | 41 ShapeRefiner::ShapeRefiner(int graph_def_version, in ShapeRefiner() function in tensorflow::ShapeRefiner
47 ShapeRefiner::ShapeRefiner(const VersionDef& versions, in ShapeRefiner() function in tensorflow::ShapeRefiner
49 : ShapeRefiner(versions.producer(), ops) {} in ShapeRefiner()
51 ShapeRefiner::~ShapeRefiner() { in ~ShapeRefiner()
65 Status InferShapesForFunctionSubNode(const Node* node, ShapeRefiner* refiner, in InferShapesForFunctionSubNode()
131 Status ShapeRefiner::InferShapesForFunction( in InferShapesForFunction()
210 Status ShapeRefiner::AddNode(const Node* node) { in AddNode()
279 Status ShapeRefiner::SetShape(const Node* node, int output_port, in SetShape()
307 Status ShapeRefiner::UpdateNode(const Node* node, bool relax, bool* refined) { in UpdateNode()
406 Status ShapeRefiner::EvaluateConstantTensorForEdge(const Node* node, in EvaluateConstantTensorForEdge()
[all …]
|
| D | shape_refiner.h | 89 class ShapeRefiner {
91 ShapeRefiner(int graph_def_version, const OpRegistryInterface* ops);
94 ShapeRefiner(const VersionDef& versions, const OpRegistryInterface* ops);
96 ~ShapeRefiner();
301 TF_DISALLOW_COPY_AND_ASSIGN(ShapeRefiner);
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | scope_internal.h | 23 class ShapeRefiner; variable
29 Scope NewInternalScope(Graph* graph, Status* status, ShapeRefiner* refiner);
44 const std::shared_ptr<ShapeRefiner>& refiner);
64 Impl(Graph* graph, Status* status, NameMap* name_map, ShapeRefiner* refiner,
95 std::shared_ptr<ShapeRefiner> refiner_ = nullptr;
|
| D | scope.cc | 40 ShapeRefiner* refiner, bool disable_shape_inference) in Impl()
52 const std::shared_ptr<ShapeRefiner>& refiner) in Impl()
63 ShapeRefiner* refiner = in NewRootScope()
64 new ShapeRefiner(graph->versions(), graph->op_registry()); in NewRootScope()
71 ShapeRefiner* refiner = in DisabledShapeInferenceScope()
72 new ShapeRefiner(graph->versions(), graph->op_registry()); in DisabledShapeInferenceScope()
434 static Scope NewScope(Graph* graph, Status* status, ShapeRefiner* refiner) { in NewScope()
445 std::shared_ptr<ShapeRefiner>(refiner, [](ShapeRefiner*) {}))); in NewScope() argument
449 Scope NewInternalScope(Graph* graph, Status* status, ShapeRefiner* refiner) { in NewInternalScope()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_constructor_test.cc | 73 ShapeRefiner* refiner = nullptr, in ExpectError()
97 ShapeRefiner* refiner = nullptr, in ExpectOK()
1167 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1207 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1270 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1335 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1367 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1388 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1410 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1435 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
[all …]
|
| D | graph_constructor.h | 25 class ShapeRefiner; variable
178 ShapeRefiner* refiner,
|
| D | graph_constructor.cc | 120 const FunctionDefLibrary* library, Graph* g, ShapeRefiner* refiner, in Construct()
141 ShapeRefiner* refiner, in GraphConstructor()
240 ShapeRefiner* refiner_;
1171 ShapeRefiner refiner(gdef.versions().producer(), g->op_registry()); in ConvertGraphDefToGraph()
1180 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, g->op_registry()); in ConvertNodeDefsToGraph()
1193 Graph* g, ShapeRefiner* refiner, in ImportGraphDef()
1224 ShapeRefiner default_refiner(gdef.versions().producer(), g->op_registry()); in ImportGraphDef()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.h | 143 Graph* graph, ShapeRefiner* shape_refiner);
146 const ShapeRefiner& shape_refiner,
|
| D | remote_fused_graph_execute_utils.cc | 441 Graph* graph, ShapeRefiner* shape_refiner) { in PropagateShapeInference()
487 const Graph& graph, const ShapeRefiner& shape_refiner, in BuildTensorShapeMapFromGraph()
587 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in BuildAndAddTensorShapes()
730 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in ClusterizeNodes()
835 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in BuildClusterSubgraphDef()
889 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in BuildClusterByBorder()
961 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in FuseCluster()
|
| D | remote_fused_graph_execute_utils_test.cc | 348 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in TEST()
467 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in TEST()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_internal.h | 81 tensorflow::ShapeRefiner refiner GUARDED_BY(mu);
|
| D | c_api.cc | 2129 tensorflow::ShapeRefiner* dst_refiner, in CopyGraph()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 29 class ShapeRefiner; variable
653 friend class ::tensorflow::ShapeRefiner;
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 388 explicit SymbolicShapeRefiner(ShapeRefiner* shape_refiner) in SymbolicShapeRefiner()
582 ShapeRefiner* shape_refiner_;
926 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in InferStatically()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.cc | 550 const Graph& graph_in, const ShapeRefiner& shape_refiner,
573 ShapeRefiner* shape_refiner,
1689 const Graph& graph_in, const ShapeRefiner& shape_refiner, in DoStaticShapeInferenceForOutsideCompilationSend()
1864 ShapeRefiner* shape_refiner, in MakeGraphForOutsideCompilationSends()
1908 ShapeRefiner shape_refiner(graph_out->versions(), graph_out->op_registry()); in GetShapeInfoForOutsideCompilationSends()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_constants_lib.cc | 194 ShapeRefiner shape_refiner(input_graph.versions(), input_graph.op_registry()); in FoldConstants()
|
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| D | range_coder_ops_test.cc | 338 ShapeRefiner refiner{g.versions().producer(), g.op_registry()}; in TEST_F()
|