| /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() 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/ |
| D | shape_inference.cc | 44 ShapeRefiner* shape_refiner) { in PropagateShapes() argument
54 Status status = shape_refiner->AddNode(n); in PropagateShapes()
59 shape_inference::InferenceContext* context = shape_refiner->GetContext(n); in PropagateShapes()
74 shape_refiner->GetContext(n); in PropagateShapes()
90 TF_RETURN_IF_ERROR(shape_refiner->SetShape(n, 0, handle)); in PropagateShapes()
98 Status StoreOutputShapes(const Graph& graph, const ShapeRefiner& shape_refiner, in StoreOutputShapes() argument
101 shape_inference::InferenceContext* context = shape_refiner.GetContext(node); in StoreOutputShapes()
138 ShapeRefiner shape_refiner(graph->versions(), graph->op_registry()); in InferShapes() local
139 shape_refiner.set_require_shape_inference_fns(false); in InferShapes()
152 TF_RETURN_IF_ERROR(PropagateShapes(*graph, arg_shapes, &shape_refiner)); in InferShapes()
[all …]
|
| D | xla_fusion_optimizer.cc | 139 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in Optimize() local
140 shape_refiner.set_require_shape_inference_fns(false); in Optimize()
141 shape_refiner.set_disable_constant_propagation(true); in Optimize()
150 ImportGraphDef(options, item.graph, &graph, &shape_refiner)); 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() argument
1982 shape_refiner.GetContext(src_node); in DoStaticShapeInferenceForOutsideCompilationSend()
2343 BackEdgeHelper* back_edge_helper, ShapeRefiner* shape_refiner, in MakeGraphForOutsideCompilationSends() argument
2373 shape_refiner->set_require_shape_inference_fns(false); in MakeGraphForOutsideCompilationSends()
2380 Status status = shape_refiner->AddNode(node); in MakeGraphForOutsideCompilationSends()
2393 ShapeRefiner shape_refiner(graph_out->versions(), graph_out->op_registry()); in GetShapeInfoForOutsideCompilationSends() local
2396 *graph_out, &pruned_graph, &back_edge_helper, &shape_refiner, in GetShapeInfoForOutsideCompilationSends()
2426 *pruned_graph, back_edge_helper, shape_refiner, recv_at_host_names, in GetShapeInfoForOutsideCompilationSends()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.h | 134 SymbolicShapeRefiner* shape_refiner, const NodeDef* qnode,
144 SymbolicShapeRefiner* shape_refiner, bool* new_shapes);
148 SymbolicShapeRefiner* shape_refiner,
153 Status UpdateMerge(SymbolicShapeRefiner* shape_refiner, const NodeDef* node,
156 static Status UpdateEnter(SymbolicShapeRefiner* shape_refiner,
160 Status UpdateShapes(SymbolicShapeRefiner* shape_refiner,
167 SymbolicShapeRefiner* shape_refiner, TopoQueue* new_shapes,
|
| D | graph_properties.cc | 1759 SymbolicShapeRefiner* shape_refiner, const NodeDef* qnode, in RelaxEnqueueShapesAndMergeTypes() argument
1776 b.shape = shape_refiner->OutputAsUnion(qnode, i, a.shape, b.shape); in RelaxEnqueueShapesAndMergeTypes()
1783 Status GraphProperties::UpdateMerge(SymbolicShapeRefiner* shape_refiner, in UpdateMerge() argument
1786 InferenceContext* ic = shape_refiner->GetContext(node); in UpdateMerge()
1789 TF_RETURN_IF_ERROR(shape_refiner->AddNode(node)); in UpdateMerge()
1790 ic = CHECK_NOTNULL(shape_refiner->GetContext(node)); in UpdateMerge()
1802 for (const GraphView::Edge fanin : shape_refiner->graph().GetFaninEdges( in UpdateMerge()
1804 InferenceContext* src_ic = shape_refiner->GetContext(fanin.src.node); in UpdateMerge()
1822 out = shape_refiner->OutputAsUnion(node, 0, input, out); in UpdateMerge()
1826 if (*new_shapes || !shape_refiner->EquivalentShapes(out, ic->output(0))) { in UpdateMerge()
[all …]
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_constants_lib.cc | 219 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/ |
| D | remote_fused_graph_execute_utils.cc | 443 Graph* graph, ShapeRefiner* shape_refiner) { in PropagateShapeInference() argument
445 auto visit = [&shape_refiner, &input_node_info_list, &status](Node* node) { in PropagateShapeInference()
457 shape_refiner->GetContext(node); in PropagateShapeInference()
464 status = shape_refiner->SetShape(node, 0, handle); in PropagateShapeInference()
476 status = shape_refiner->AddNode(node); in PropagateShapeInference()
489 const Graph& graph, const ShapeRefiner& shape_refiner, in BuildTensorShapeMapFromGraph() argument
498 shape_refiner.GetContext(node); in BuildTensorShapeMapFromGraph()
588 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in BuildAndAddTensorShapes() local
590 ImportGraphDef(opts, *graph_def, &graph, &shape_refiner)); in BuildAndAddTensorShapes()
592 &graph, &shape_refiner)); in BuildAndAddTensorShapes()
[all …]
|
| D | remote_fused_graph_execute_utils_test.cc | 349 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in TEST() local
350 Status status = ImportGraphDef(opts, def, &graph, &shape_refiner); in TEST()
352 def, inputs, &graph, &shape_refiner) in TEST()
355 graph, shape_refiner, &tensor_shape_map) in TEST()
468 ShapeRefiner shape_refiner(graph.versions(), graph.op_registry()); in TEST() local
469 TF_ASSERT_OK(ImportGraphDef({}, def, &graph, &shape_refiner)); in TEST()
|
| D | remote_fused_graph_execute_utils.h | 152 Graph* graph, ShapeRefiner* shape_refiner);
155 const ShapeRefiner& shape_refiner,
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 1422 "common_runtime/shape_refiner.h",
2983 "common_runtime/shape_refiner.h",
2997 "common_runtime/shape_refiner.cc",
2998 "common_runtime/shape_refiner.h",
|