| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.h | 116 Graph* graph, ShapeRefiner* shape_refiner);
120 const ShapeRefiner& shape_refiner, const Node& node,
124 void RegisterConstantNode(const ShapeRefiner& shape_refiner,
144 const ShapeRefiner& shape_refiner);
148 const ShapeRefiner& shape_refiner, const Node& node);
152 const ShapeRefiner& shape_refiner, const Node& node);
155 const ShapeRefiner& shape_refiner, const Node& node);
158 const ShapeRefiner& shape_refiner, const Node& node);
162 const ShapeRefiner& shape_refiner, const Node& node);
166 const ShapeRefiner& shape_refiner, const Node& node);
[all …]
|
| D | graph_transferer.cc | 98 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in LoadGraphFromProto()
325 Graph* graph, ShapeRefiner* shape_refiner) { in TransformGraphToAddAggregatedInputNode()
423 const ShapeRefiner& shape_refiner, const Node& node, in RegisterNode()
456 void GraphTransferer::RegisterConstantNode(const ShapeRefiner& shape_refiner, in RegisterConstantNode()
605 const ShapeRefiner& shape_refiner) { in IsNodeFlattenReshape()
645 const ShapeRefiner& shape_refiner, const Node& node) { in RegisterNodeWithPaddingAndStrides()
679 const ShapeRefiner& shape_refiner, const Node& node) { in RegisterNodeWithRank()
712 const ShapeRefiner& shape_refiner, const Node& node) { in RegisterPadNode()
786 const ShapeRefiner& shape_refiner, const Node& node) { in RegisterInputNode()
804 const ShapeRefiner& shape_refiner, const Node& node) { in RegisterFlattenNode()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 42 ShapeRefiner::ShapeRefiner(int graph_def_version, in ShapeRefiner() function in tensorflow::ShapeRefiner
48 ShapeRefiner::ShapeRefiner(const VersionDef& versions, in ShapeRefiner() function in tensorflow::ShapeRefiner
50 : ShapeRefiner(versions.producer(), ops) {} in ShapeRefiner()
52 ShapeRefiner::~ShapeRefiner() { in ~ShapeRefiner()
66 Status InferShapesForFunctionSubNode(const Node* node, ShapeRefiner* refiner, in InferShapesForFunctionSubNode()
132 Status ShapeRefiner::InferShapesForFunction( in InferShapesForFunction()
211 Status ShapeRefiner::AddNode(const Node* node) { in AddNode()
279 Status ShapeRefiner::SetShape(const Node* node, int output_port, in SetShape()
312 Status ShapeRefiner::UpdateNode(const Node* node, bool relax, bool* refined) { in UpdateNode()
416 Status ShapeRefiner::EvaluateConstantTensorForEdge(const Node* node, in EvaluateConstantTensorForEdge()
[all …]
|
| D | shape_refiner_test.cc | 52 return ShapeRefiner::SameDefinedShape(c, s0, s1); in SameDefinedShape()
59 return ShapeRefiner::IsUpdatedShapesOrTypes(c, existing, updated); in IsUpdatedShapesOrTypes()
62 static constexpr int64 kMaxTensorSize = ShapeRefiner::kMaxTensorSize;
84 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TestStridedSlice()
129 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
136 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
153 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
168 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
185 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
230 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
[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);
|
| D | eval_const_tensor.h | 27 class ShapeRefiner; variable
57 OutputTensor tensor, const ShapeRefiner& refiner,
|
| D | eval_const_tensor.cc | 39 const ShapeRefiner& refiner, in TryToInferTensorOutputFromInputShapes()
127 const Node& target_node, const ShapeRefiner& refiner, in ExtractConstantSubgraph()
308 Status EvaluateConstantTensor(OutputTensor tensor, const ShapeRefiner& refiner, in EvaluateConstantTensor()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | scope_internal.h | 23 class ShapeRefiner; variable
31 Scope NewInternalScope(Graph* graph, Status* status, ShapeRefiner* refiner);
45 const std::shared_ptr<ShapeRefiner>& refiner);
67 Impl(Graph* graph, Status* status, NameMap* name_map, ShapeRefiner* refiner,
100 std::shared_ptr<ShapeRefiner> refiner_ = nullptr;
|
| D | scope.cc | 45 ShapeRefiner* refiner, bool disable_shape_inference) in Impl()
57 const std::shared_ptr<ShapeRefiner>& refiner) in Impl()
68 ShapeRefiner* refiner = in NewRootScope()
69 new ShapeRefiner(graph->versions(), graph->op_registry()); in NewRootScope()
76 ShapeRefiner* refiner = in DisabledShapeInferenceScope()
77 new ShapeRefiner(graph->versions(), graph->op_registry()); in DisabledShapeInferenceScope()
509 static Scope NewScope(Graph* graph, Status* status, ShapeRefiner* refiner) { in NewScope()
526 std::shared_ptr<ShapeRefiner>(refiner, [](ShapeRefiner*) {}))); in NewScope() argument
530 Scope NewInternalScope(Graph* graph, Status* status, ShapeRefiner* refiner) { in NewInternalScope()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_constructor.h | 25 class ShapeRefiner; variable
181 ShapeRefiner* refiner,
|
| D | graph_constructor_test.cc | 73 ShapeRefiner* refiner = nullptr, in ExpectError()
97 ShapeRefiner* refiner = nullptr, in ExpectOK()
1172 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1212 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1275 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1340 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1372 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1393 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1415 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1440 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
[all …]
|
| D | graph_constructor.cc | 126 const FunctionDefLibrary* library, Graph* g, ShapeRefiner* refiner, in Construct()
147 ShapeRefiner* refiner, in GraphConstructor()
250 ShapeRefiner* refiner_;
1209 ShapeRefiner refiner(gdef.versions().producer(), g->op_registry()); in ConvertGraphDefToGraph()
1218 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, g->op_registry()); in ConvertNodeDefsToGraph()
1231 Graph* g, ShapeRefiner* refiner, in ImportGraphDef()
1262 ShapeRefiner default_refiner(gdef.versions().producer(), g->op_registry()); in ImportGraphDef()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference.cc | 44 ShapeRefiner* shape_refiner) { in PropagateShapes()
98 Status StoreOutputShapes(const Graph& graph, const ShapeRefiner& shape_refiner, in StoreOutputShapes()
138 ShapeRefiner shape_refiner(graph->versions(), graph->op_registry()); in InferShapes()
|
| D | xla_fusion_optimizer.cc | 139 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in Optimize()
|
| D | encapsulate_subgraphs_pass.cc | 613 const ShapeRefiner& shape_refiner,
636 BackEdgeHelper* back_edge_helper, ShapeRefiner* shape_refiner,
1920 const ShapeRefiner& shape_refiner, in DoStaticShapeInferenceForOutsideCompilationSend()
2343 BackEdgeHelper* back_edge_helper, ShapeRefiner* shape_refiner, in MakeGraphForOutsideCompilationSends()
2393 ShapeRefiner shape_refiner(graph_out->versions(), graph_out->op_registry()); in GetShapeInfoForOutsideCompilationSends()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.h | 152 Graph* graph, ShapeRefiner* shape_refiner);
155 const ShapeRefiner& shape_refiner,
|
| D | remote_fused_graph_execute_utils.cc | 443 Graph* graph, ShapeRefiner* shape_refiner) { in PropagateShapeInference()
489 const Graph& graph, const ShapeRefiner& shape_refiner, in BuildTensorShapeMapFromGraph()
588 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in BuildAndAddTensorShapes()
731 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in ClusterizeNodes()
836 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in BuildClusterSubgraphDef()
890 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 | 349 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in TEST()
468 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in TEST()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_internal.h | 85 tensorflow::ShapeRefiner refiner GUARDED_BY(mu);
|
| D | c_api.cc | 2248 tensorflow::ShapeRefiner* dst_refiner, in CopyGraph()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 29 class ShapeRefiner; variable
684 friend class ::tensorflow::ShapeRefiner;
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_constants_lib.cc | 219 ShapeRefiner shape_refiner(input_graph.versions(), input_graph.op_registry()); in FoldConstants()
|
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| D | range_coder_ops_test.cc | 337 ShapeRefiner refiner{g.versions().producer(), g.op_registry()}; in TEST_F()
|