Home
last modified time | relevance | path

Searched refs:graph_properties_ (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.h100 graph_properties_ = &graph_properties; in SetGraphProperties()
101 CHECK(graph_properties_); in SetGraphProperties()
105 const GraphProperties* graph_properties_;
Darithmetic_optimizer.h137 std::unique_ptr<GraphProperties> graph_properties_; variable
Dlayout_optimizer.cc409 : graph_properties_(graph_properties), in GraphProcessor()
470 const GraphProperties& graph_properties_; member in tensorflow::grappler::__anon49c8f8520111::GraphProcessor
657 graph_properties_.GetInputProperties(node_->name())[pos].dtype(); in AddLayoutTransposeToInputs()
695 graph_properties_.GetOutputProperties(node_->name())[input_port] in AddTransformToOutputs()
2067 OptimizeContext opt_cxt(graph_, node, node_map_, graph_properties_, in Expand()
2115 OptimizeContext opt_cxt(graph_, node, node_map_, graph_properties_, in Expand()
Dscoped_allocator_optimizer.cc453 CHECK(graph_properties_); in AnalyzeInputs()
455 CheckTypesAndGetShapes(*graph_properties_, ops, dtype, input_shapes)); in AnalyzeInputs()
457 GetInputs(sa_opti, invocation_count, graph, *graph_properties_, in AnalyzeInputs()
Darithmetic_optimizer.cc3672 graph_properties_.get(), node_map_.get(), in SimplifyArithmeticOps()
3811 graph_properties_.reset(new GraphProperties(optimized_item)); in Optimize()
3814 graph_properties_->InferStatically(assume_valid_feeds, in Optimize()
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc420 graph_properties_ = absl::make_unique<GraphProperties>(*item); in Init()
424 TF_RETURN_IF_ERROR(graph_properties_->InferStatically( in Init()
427 TF_RETURN_IF_ERROR(graph_properties_->InferDynamically(cluster_)); in Init()
622 graph_properties_->GetOutputProperties(NodeName(input_source_name)); in MaybeUpdateInputOutput()
789 graph_properties_->GetInputProperties(node->name()); in GetNodeStateOrCreateIt()
791 graph_properties_->GetOutputProperties(node->name()); in GetNodeStateOrCreateIt()
793 graph_properties_->CheckShapeIncompatible(node->name()); in GetNodeStateOrCreateIt()
Dvirtual_scheduler.h438 std::unique_ptr<GraphProperties> graph_properties_; // Initialized in Init(). variable
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h427 const grappler::GraphProperties& graph_properties_; variable
Dconvert_nodes.cc1140 : graph_properties_(graph_properties), in TrtNodeValidator()
1164 if (!graph_properties_.HasOutputProperties(node_def.name())) { in ConvertToTensorOrWeights()
1170 graph_properties_.GetOutputProperties(node_def.name()); in ConvertToTensorOrWeights()