Searched refs:graph_properties_ (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.h | 86 graph_properties_ = &graph_properties; in SetGraphProperties()
87 CHECK(graph_properties_); in SetGraphProperties()
91 const GraphProperties* graph_properties_;
|
| D | arithmetic_optimizer.h | 133 std::unique_ptr<GraphProperties> graph_properties_; variable
|
| D | layout_optimizer.cc | 401 : graph_properties_(graph_properties), in GraphProcessor()
462 const GraphProperties& graph_properties_; member in tensorflow::grappler::__anonbd4a17fb0111::GraphProcessor
645 graph_properties_.GetInputProperties(node_->name())[pos].dtype(); in AddLayoutTransposeToInputs()
683 graph_properties_.GetOutputProperties(node_->name())[input_port] in AddTransformToOutputs()
2009 OptimizeContext opt_cxt(graph_, node, node_map_, graph_properties_, in Expand()
2061 OptimizeContext opt_cxt(graph_, node, node_map_, graph_properties_, in Expand()
|
| D | scoped_allocator_optimizer.cc | 285 CHECK(graph_properties_); in AnalyzeInputs()
287 CheckTypesAndGetShapes(*graph_properties_, ops, dtype, input_shapes)); in AnalyzeInputs()
|
| D | arithmetic_optimizer.cc | 3535 graph_properties_.get(), node_map_.get(), in SimplifyArithmeticOps()
3676 graph_properties_.reset(new GraphProperties(optimized_item)); in Optimize()
3678 const Status status = graph_properties_->InferStatically(assume_valid_feeds); in Optimize()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler.cc | 276 graph_properties_ = absl::make_unique<GraphProperties>(*item); in Init()
301 TF_RETURN_IF_ERROR(graph_properties_->InferStatically( in Init()
304 TF_RETURN_IF_ERROR(graph_properties_->InferDynamically(cluster_)); in Init()
500 graph_properties_->GetOutputProperties(NodeName(input_source_name)); in MaybeUpdateInputOutput()
664 graph_properties_->GetInputProperties(node->name()); in GetNodeStateOrCreateIt()
666 graph_properties_->GetOutputProperties(node->name()); in GetNodeStateOrCreateIt()
|
| D | virtual_scheduler.h | 350 std::unique_ptr<GraphProperties> graph_properties_; // Initialized in Init(). variable
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.h | 52 const grappler::GraphProperties& graph_properties_; variable
|
| D | convert_graph.cc | 71 : graph_properties_(graph_properties), precision_mode_(precision_mode) {} in TrtCandidateSelector()
83 precision_mode_, graph_properties_); in IsTensorRTCandidate()
|