Home
last modified time | relevance | path

Searched refs:shape_refiner (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.h116 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 …]
Dgraph_transferer.cc98 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in LoadGraphFromProto() local
99 Status status = ImportGraphDef({}, graph_def, &graph, &shape_refiner); in LoadGraphFromProto()
106 graph_def, input_node_info_list, &graph, &shape_refiner); in LoadGraphFromProto()
113 input_node_info_list, &graph, &shape_refiner)); in LoadGraphFromProto()
141 ops_definitions, shape_refiner, *node, false, input_node_info_list, in LoadGraphFromProto()
325 Graph* graph, ShapeRefiner* shape_refiner) { in TransformGraphToAddAggregatedInputNode() argument
357 shape_refiner->UpdateNode(input_node, false /* relax */, &refined)); in TransformGraphToAddAggregatedInputNode()
360 shape_refiner->GetContext(input_node); in TransformGraphToAddAggregatedInputNode()
365 TF_RETURN_IF_ERROR(shape_refiner->SetShape(input_node, i, handle)); in TransformGraphToAddAggregatedInputNode()
402 shape_refiner->UpdateNode(created_node, false /* relax */, &refined)); in TransformGraphToAddAggregatedInputNode()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference.cc48 ShapeRefiner* shape_refiner) { in PropagateShapes() argument
65 Status status = shape_refiner->AddNode(n); in PropagateShapes()
70 shape_inference::InferenceContext* context = shape_refiner->GetContext(n); in PropagateShapes()
85 shape_refiner->GetContext(n); in PropagateShapes()
101 TF_RETURN_IF_ERROR(shape_refiner->SetShape(n, 0, handle)); in PropagateShapes()
110 shape_inference::InferenceContext* context = shape_refiner->GetContext(n); in PropagateShapes()
189 shape_refiner->GetContext(n); in PropagateShapes()
209 Status StoreOutputShapes(const Graph& graph, const ShapeRefiner& shape_refiner, in StoreOutputShapes() argument
212 shape_inference::InferenceContext* context = shape_refiner.GetContext(node); in StoreOutputShapes()
249 ShapeRefiner shape_refiner(graph->versions(), graph->op_registry()); in InferShapes() local
[all …]
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.h164 SymbolicShapeRefiner* shape_refiner, const NodeDef* qnode,
174 SymbolicShapeRefiner* shape_refiner, bool* new_shapes);
178 SymbolicShapeRefiner* shape_refiner,
183 Status UpdateMerge(SymbolicShapeRefiner* shape_refiner, const NodeDef* node,
186 static Status UpdateEnter(SymbolicShapeRefiner* shape_refiner,
190 Status UpdateShapes(SymbolicShapeRefiner* shape_refiner,
197 SymbolicShapeRefiner* shape_refiner, TopoQueue* new_shapes,
Dgraph_properties.cc2192 SymbolicShapeRefiner* shape_refiner, const NodeDef* qnode, in RelaxEnqueueShapesAndMergeTypes() argument
2209 b.shape = shape_refiner->OutputAsUnion(qnode, i, a.shape, b.shape); in RelaxEnqueueShapesAndMergeTypes()
2216 Status GraphProperties::UpdateMerge(SymbolicShapeRefiner* shape_refiner, in UpdateMerge() argument
2219 InferenceContext* ic = shape_refiner->GetContext(node); in UpdateMerge()
2222 TF_RETURN_IF_ERROR(shape_refiner->AddNode(node)); in UpdateMerge()
2223 ic = CHECK_NOTNULL(shape_refiner->GetContext(node)); in UpdateMerge()
2235 for (const GraphView::Edge fanin : shape_refiner->graph().GetFaninEdges( in UpdateMerge()
2237 InferenceContext* src_ic = shape_refiner->GetContext(fanin.src.node); in UpdateMerge()
2255 out = shape_refiner->OutputAsUnion(node, 0, input, out); in UpdateMerge()
2259 if (*new_shapes || !shape_refiner->EquivalentShapes(out, ic->output(0))) { in UpdateMerge()
[all …]
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_lib.cc219 ShapeRefiner shape_refiner(input_graph.versions(), input_graph.op_registry()); in FoldConstants() local
220 shape_refiner.set_require_shape_inference_fns(false); in FoldConstants()
221 shape_refiner.set_disable_constant_propagation(false); in FoldConstants()
222 shape_refiner.set_function_library_for_shape_inference( in FoldConstants()
235 ImportGraphDef({}, cleaned_graph_def, &input_graph, &shape_refiner)); in FoldConstants()
238 ImportGraphDef({}, input_graph_def, &input_graph, &shape_refiner)); in FoldConstants()
257 auto ctx = shape_refiner.GetContext(node); in FoldConstants()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc444 Graph* graph, ShapeRefiner* shape_refiner) { in PropagateShapeInference() argument
446 auto visit = [&shape_refiner, &input_node_info_list, &status](Node* node) { in PropagateShapeInference()
458 shape_refiner->GetContext(node); in PropagateShapeInference()
465 status = shape_refiner->SetShape(node, 0, handle); in PropagateShapeInference()
477 status = shape_refiner->AddNode(node); in PropagateShapeInference()
490 const Graph& graph, const ShapeRefiner& shape_refiner, in BuildTensorShapeMapFromGraph() argument
499 shape_refiner.GetContext(node); in BuildTensorShapeMapFromGraph()
589 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in BuildAndAddTensorShapes() local
591 ImportGraphDef(opts, *graph_def, &graph, &shape_refiner)); in BuildAndAddTensorShapes()
593 &graph, &shape_refiner)); in BuildAndAddTensorShapes()
[all …]
Dremote_fused_graph_execute_utils_test.cc348 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in TEST() local
349 Status status = ImportGraphDef(opts, def, &graph, &shape_refiner); in TEST()
351 def, inputs, &graph, &shape_refiner) in TEST()
354 graph, shape_refiner, &tensor_shape_map) in TEST()
467 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in TEST() local
468 TF_ASSERT_OK(ImportGraphDef({}, def, &graph, &shape_refiner)); in TEST()
Dremote_fused_graph_execute_utils.h152 Graph* graph, ShapeRefiner* shape_refiner);
155 const ShapeRefiner& shape_refiner,
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dhost_compute_ops.cc298 ShapeRefiner shape_refiner{graph->versions().producer(), in InferOutputShapes() local
303 TF_RETURN_IF_ERROR(shape_refiner.AddNode(node)); in InferOutputShapes()
328 shape_refiner.GetContext(node); in InferOutputShapes()
344 shape_refiner.GetContext(node); in InferOutputShapes()
/external/tensorflow/tensorflow/core/common_runtime/
DBUILD67 "shape_refiner.h",
172 "shape_refiner.h",
739 # "shape_refiner", because there are circular dependencies between these
747 "shape_refiner.cc",
754 "shape_refiner.h",
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond.cc1521 ShapeRefiner shape_refiner{graph_->versions().producer(), in FunctionalizeInternal() local
1526 if (!shape_refiner.AddNode(node).ok()) { in FunctionalizeInternal()
1574 shape_refiner); in FunctionalizeInternal()