| /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()
68 Status ShapeRefiner::InferShapesForFunctionSubNode( in InferShapesForFunctionSubNode()
140 Status ShapeRefiner::InferShapesForFunction( in InferShapesForFunction()
190 Status ShapeRefiner::AddNode(const Node* node) { in AddNode()
194 Status ShapeRefiner::AddNodeInternal( in AddNodeInternal()
257 Status ShapeRefiner::SetShape(const Node* node, int output_port, in SetShape()
290 Status ShapeRefiner::UpdateNode(const Node* node, bool relax, bool* refined) { in UpdateNode()
[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()
170 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
215 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
233 ShapeRefiner m(TF_GRAPH_DEF_VERSION, OpRegistry::Global()); in TEST_F()
[all …]
|
| D | shape_refiner.h | 73 class ShapeRefiner {
75 ShapeRefiner(int graph_def_version, const OpRegistryInterface* ops);
78 ShapeRefiner(const VersionDef& versions, const OpRegistryInterface* ops);
80 ~ShapeRefiner();
329 TF_DISALLOW_COPY_AND_ASSIGN(ShapeRefiner);
|
| D | eval_const_tensor.h | 27 class ShapeRefiner; variable
60 OutputTensor tensor, const ShapeRefiner& refiner,
|
| D | graph_constructor.h | 25 class ShapeRefiner; variable
193 ShapeRefiner* refiner,
|
| D | graph_constructor_test.cc | 82 ShapeRefiner* refiner = nullptr, in ExpectError()
106 ShapeRefiner* refiner = nullptr, in ExpectOK()
1176 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1216 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1279 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1344 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1376 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1397 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1419 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1444 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
[all …]
|
| D | graph_constructor.cc | 160 const FunctionDefLibrary* library, Graph* g, ShapeRefiner* refiner,
167 ShapeRefiner* refiner, std::vector<std::pair<Node*, int>>* return_tensors,
172 GraphConstructor(const Options& opts, Graph* g, ShapeRefiner* refiner, in GraphConstructor()
301 ShapeRefiner* refiner_;
405 const FunctionDefLibrary* library, Graph* g, ShapeRefiner* refiner, in NodeDefCopyingGraphConstructor()
433 ShapeRefiner* refiner, std::vector<std::pair<Node*, int>>* return_tensors, in NodeDefMovingGraphConstructor()
491 const FunctionDefLibrary* library, Graph* g, ShapeRefiner* refiner, in Construct()
512 const Options& opts, GraphDef&& graph_def, Graph* g, ShapeRefiner* refiner, in Construct()
1456 ShapeRefiner refiner(gdef.versions().producer(), g->op_registry()); in ConvertGraphDefToGraph()
1465 ShapeRefiner refiner(gdef.versions().producer(), g->op_registry()); in ConvertGraphDefToGraph()
[all …]
|
| D | eval_const_tensor.cc | 40 const ShapeRefiner& refiner, in TryToInferTensorOutputFromInputShapes()
142 const Node& target_node, const ShapeRefiner& refiner, in ExtractConstantSubgraph()
348 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/compiler/jit/ |
| D | shape_inference.cc | 48 ShapeRefiner* shape_refiner) { in PropagateShapes()
209 Status StoreOutputShapes(const Graph& graph, const ShapeRefiner& shape_refiner, in StoreOutputShapes()
249 ShapeRefiner shape_refiner(graph->versions(), graph->op_registry()); in InferShapes()
|
| /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 | 444 Graph* graph, ShapeRefiner* shape_refiner) { in PropagateShapeInference()
490 const Graph& graph, const ShapeRefiner& shape_refiner, in BuildTensorShapeMapFromGraph()
589 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in BuildAndAddTensorShapes()
732 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in ClusterizeNodes()
838 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in BuildClusterSubgraphDef()
892 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in BuildClusterByBorder()
963 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 | 77 tensorflow::ShapeRefiner refiner TF_GUARDED_BY(mu);
|
| D | c_api.cc | 1841 tensorflow::ShapeRefiner* dst_refiner, in CopyGraph()
|
| /external/tensorflow/tensorflow/tools/def_file_filter/ |
| D | symbols_pybind.txt | 222 tensorflow::ShapeRefiner::~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/compiler/tf2xla/ |
| D | functionalize_cond.cc | 303 StateMap* cond_state_map, const ShapeRefiner& refiner);
357 const ShapeRefiner& refiner_;
391 StateMap* cond_state_map, const ShapeRefiner& refiner) in Conditional()
1521 ShapeRefiner shape_refiner{graph_->versions().producer(), in FunctionalizeInternal()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 298 ShapeRefiner shape_refiner{graph->versions().producer(), in InferOutputShapes()
|
| /external/llvm-project/mlir/docs/Rationale/ |
| D | MLIRForGraphAlgorithms.md | 239 with ShapeRefiner.
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 421 std::unique_ptr<ShapeRefiner> shape_refiner_ = nullptr;
707 shape_refiner_ = absl::make_unique<ShapeRefiner>(graph_->versions(), in AddNodesToShapeRefiner()
|