Searched refs:function_library_ (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.h | 149 function_library_ = lib; in set_function_library_for_shape_inference()
153 return function_library_ != nullptr; in function_shape_inference_supported()
321 const tensorflow::FunctionLibraryDefinition* function_library_ = nullptr; variable
|
| D | shape_refiner.cc | 152 return this->function_library_->LookUpOpDef(op, sig); in InferShapesForFunction()
156 Graph* new_graph = new Graph(function_library_); in InferShapesForFunction()
666 if (function_library_ && IsFunctionCall(*function_library_, *node)) { in RunShapeFn()
677 function_library_->Find(function.name()); in RunShapeFn()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 261 function_library_(OpRegistry::Global(), graph.library()), in FunctionOptimizerContext()
272 return function_library_; in function_library()
274 FunctionLibraryDefinition& function_library() { return function_library_; } in function_library()
361 FunctionLibraryDefinition function_library_; member in tensorflow::grappler::__anon36252d010111::FunctionOptimizerContext
|
| D | auto_mixed_precision.cc | 230 function_library_.reset( in Init()
283 TF_RETURN_IF_ERROR(function_library_->LookUpOpDef(node.op(), &op_def_ptr)); in AddNode()
351 std::unique_ptr<FunctionLibraryDefinition> function_library_; member in tensorflow::grappler::__anonab6217530111::NodeTypeAttrMap
935 function_library_(OpRegistry::Global(), graph->library()), in AutoMixedPrecisionImpl()
1001 FunctionLibraryDefinition function_library_; member in tensorflow::grappler::__anonab6217530111::AutoMixedPrecisionImpl
1440 return source_and_sink_ops.count(op) || function_library_.Find(op); in IsSourceOrSinkOp()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 621 function_library_(OpRegistry::Global(), graph.graph()->library()), in SymbolicShapeRefiner()
1276 CHECK_NOTNULL(function_library_.Find(function.name())); in AddFunction()
1279 MakeGrapplerFunctionItem(*function_def, function_library_, in AddFunction()
1321 function_library_.LookUp(function.name(), &node_ctx.op_data)); in AddNode()
2008 FunctionLibraryDefinition function_library_; member in tensorflow::grappler::SymbolicShapeRefiner
|