Home
last modified time | relevance | path

Searched refs:graph_properties (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
DBUILD38 name = "graph_properties",
39 srcs = ["graph_properties.cc"],
40 hdrs = ["graph_properties.h"],
65 ":graph_properties",
89 ":graph_properties",
230 ":graph_properties",
315 ":graph_properties",
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.h55 const GraphProperties& graph_properties);
85 void SetGraphProperties(const GraphProperties& graph_properties) { in SetGraphProperties() argument
86 graph_properties_ = &graph_properties; in SetGraphProperties()
DBUILD28 "//tensorflow/core/grappler/costs:graph_properties",
102 "//tensorflow/core/grappler/costs:graph_properties",
216 "//tensorflow/core/grappler/costs:graph_properties",
234 "//tensorflow/core/grappler/costs:graph_properties",
271 "//tensorflow/core/grappler/costs:graph_properties",
337 "//tensorflow/core/grappler/costs:graph_properties",
451 "//tensorflow/core/grappler/costs:graph_properties",
497 "//tensorflow/core/grappler/costs:graph_properties",
653 "//tensorflow/core/grappler/costs:graph_properties",
694 "//tensorflow/core/grappler/costs:graph_properties",
[all …]
Dgraph_optimizer_stage.h51 GraphProperties* graph_properties, NodeMap* node_map, in GraphOptimizerContext()
56 graph_properties(graph_properties), in GraphOptimizerContext()
63 GraphProperties* graph_properties; member
Dgraph_optimizer_stage.cc46 if (ctx.graph_properties == nullptr) { in GetTensorProperties()
60 ctx.graph_properties->GetOutputProperties(tensor_id.node()); in GetTensorProperties()
Dscoped_allocator_optimizer.cc78 Status CheckTypesAndGetShapes(const GraphProperties& graph_properties, in CheckTypesAndGetShapes() argument
89 if (!graph_properties.HasOutputProperties(n->name())) { in CheckTypesAndGetShapes()
94 graph_properties.GetOutputProperties(n->name()); in CheckTypesAndGetShapes()
683 GraphProperties graph_properties(item); in Optimize() local
686 graph_properties.InferStatically(assume_valid_feeds)); in Optimize()
690 optimized_graph, graph_properties)); in Optimize()
818 GraphDef* graph, const GraphProperties& graph_properties) { in ProcessGraphDef() argument
847 rewriter->SetGraphProperties(graph_properties); in ProcessGraphDef()
Dlayout_optimizer.cc397 GraphProcessor(const GraphProperties& graph_properties, in GraphProcessor() argument
401 : graph_properties_(graph_properties), in GraphProcessor()
471 const GraphProperties& graph_properties, in OptimizeContext()
478 graph_properties(graph_properties), in OptimizeContext()
485 const GraphProperties& graph_properties; member
494 : GraphProcessor(opt_cxt.graph_properties, opt_cxt.virtual_placer, in NodeProcessor()
1962 const GraphProperties& graph_properties, in DataLayoutOptimizer() argument
1967 : GraphProcessor(graph_properties, virtual_placer, nodes_to_preserve, in DataLayoutOptimizer()
2182 const GraphProperties& graph_properties, in Tune() argument
2184 auto status = graph_properties.AnnotateOutputShapes(output); in Tune()
[all …]
Dlayout_optimizer.h53 Status Tune(const GrapplerItem& item, const GraphProperties& graph_properties,
Dremapper.cc53 graph_properties(item), in RemapperContext()
58 GraphProperties graph_properties; member
199 ctx.graph_properties.GetInputProperties(matched.conv2d->name()); in IsGpuCompatible()
437 const auto& props = ctx.graph_properties.GetInputProperties(node->name()); in FindFusedBatchNorm()
849 TF_RETURN_IF_ERROR(ctx.graph_properties.InferStatically(false)); in Optimize()
Darithmetic_optimizer.cc1067 ctx().graph_properties->ClearOutputProperties(node->name()); in UpdateInputs()
1068 ctx().graph_properties->ClearInputProperties(node->name()); in UpdateInputs()
2467 ctx().graph_properties->GetInputProperties(node->name()).size() == 2; in IsSupported()
2472 ctx().graph_properties->GetInputProperties(node->name())[1]; in TrySimplify()
2501 ctx().graph_properties->GetInputProperties(node->name())[0]; in TrySimplify()
2503 ctx().graph_properties->GetOutputProperties(node->name())[0].shape(); in TrySimplify()
2609 if (ctx().graph_properties->GetInputProperties(input->name()).size() < 2) { in TrySimplify()
2628 ctx().graph_properties->GetInputProperties(input->name())[i]; in TrySimplifyInternal()
2630 ctx().graph_properties->GetInputProperties(input->name())[j]; in TrySimplifyInternal()
2701 if (ctx().graph_properties->GetInputProperties(node->name()).size() < 2) { in TrySimplify()
[all …]
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.h38 TrtCandidateSelector(const grappler::GraphProperties& graph_properties,
66 const grappler::GraphProperties* graph_properties = nullptr; member
Dconvert_graph_test.cc98 grappler::GraphProperties graph_properties(item); in TEST() local
99 TF_EXPECT_OK(graph_properties.InferStatically(true)); in TEST()
103 TrtCandidateSelector selector(graph_properties, precision_mode); in TEST()
Dconvert_nodes.h136 const Graph* graph, const grappler::GraphProperties& graph_properties,
386 const grappler::GraphProperties& graph_properties);
403 const grappler::GraphProperties& graph_properties,
Dconvert_graph.cc69 const grappler::GraphProperties& graph_properties, in TrtCandidateSelector() argument
71 : graph_properties_(graph_properties), precision_mode_(precision_mode) {} in TrtCandidateSelector()
196 const grappler::GraphProperties& graph_properties, in GetEngineInfo() argument
336 g, graph_properties, subgraph_nodes, &info->connections, in GetEngineInfo()
817 TrtCandidateSelector candidate_selector(*params.graph_properties, in ConvertAfterShapes()
847 GetEngineInfo(&graph, *params.graph_properties, curr_segment.first, in ConvertAfterShapes()
Dconvert_nodes_test.cc403 const grappler::GraphProperties& graph_properties, in ConvertToTensorOrWeights() argument
406 node_def, output_port, graph_properties, tensor_or_weights); in ConvertToTensorOrWeights()
427 grappler::GraphProperties graph_properties(item); in TEST_F() local
429 graph_properties, &output)); in TEST_F()
443 grappler::GraphProperties graph_properties(item); in TEST_F() local
444 TF_EXPECT_OK(graph_properties.InferStatically(true)); in TEST_F()
447 graph_properties, output); in TEST_F()
481 grappler::GraphProperties graph_properties(item); in TEST_F() local
495 graph_properties), in TEST_F()
501 graph_properties)); in TEST_F()
[all …]
Dconvert_nodes.cc233 void GetOutputProperties(const grappler::GraphProperties& graph_properties, in GetOutputProperties() argument
236 if (graph_properties.HasOutputProperties(node->name())) { in GetOutputProperties()
237 auto output_params = graph_properties.GetOutputProperties(node->name()); in GetOutputProperties()
247 void GetInputProperties(const grappler::GraphProperties& graph_properties, in GetInputProperties() argument
250 if (graph_properties.HasInputProperties(node->name())) { in GetInputProperties()
251 auto input_params = graph_properties.GetInputProperties(node->name()); in GetInputProperties()
895 const grappler::GraphProperties& graph_properties, in ConvertToTensorOrWeights() argument
911 if (!graph_properties.HasOutputProperties(node_def.name())) { in ConvertToTensorOrWeights()
917 graph_properties.GetOutputProperties(node_def.name()); in ConvertToTensorOrWeights()
939 const grappler::GraphProperties& graph_properties) { in ValidateNode() argument
[all …]
Dtrt_optimization_pass.cc262 cp.graph_properties = &static_graph_properties; in Optimize()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/
DBUILD70 "//tensorflow/core/grappler/costs:graph_properties",
271 "//tensorflow/core/grappler/costs:graph_properties",
330 "//tensorflow/core/grappler/costs:graph_properties",
/external/tensorflow/tensorflow/python/
DBUILD317 "//tensorflow/core/grappler/costs:graph_properties",