/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.h | 25 class ShapeRefiner; variable 178 ShapeRefiner* refiner,
|
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.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/tools/graph_transforms/ |
D | fold_constants_lib.cc | 194 ShapeRefiner shape_refiner(input_graph.versions(), input_graph.op_registry()); in FoldConstants()
|
/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/contrib/coder/kernels/ |
D | range_coder_ops_test.cc | 338 ShapeRefiner refiner{g.versions().producer(), g.op_registry()}; in TEST_F()
|