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 | 169 return this->function_library_->LookUpOpDef(op, sig); in InferShapesForFunction()
173 Graph* new_graph = new Graph(function_library_); in InferShapesForFunction()
687 if (function_library_ && IsFunctionCall(*function_library_, *node)) { in RunShapeFn()
698 function_library_->Find(function.name()); in RunShapeFn()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 264 function_library_(OpRegistry::Global(), graph.library()), in FunctionOptimizerContext()
275 return function_library_; in function_library()
277 FunctionLibraryDefinition& function_library() { return function_library_; } in function_library()
364 FunctionLibraryDefinition function_library_; member in tensorflow::grappler::__anon36c8eb5f0111::FunctionOptimizerContext
|
| D | auto_mixed_precision.cc | 293 function_library_.reset( in Init()
349 TF_RETURN_IF_ERROR(function_library_->LookUpOpDef(node.op(), &op_def_ptr)); in AddNode()
417 std::unique_ptr<FunctionLibraryDefinition> function_library_; member in tensorflow::grappler::__anon63eee5310111::NodeTypeAttrMap
1043 function_library_(OpRegistry::Global(), graph->library()), in AutoMixedPrecisionImpl()
1131 FunctionLibraryDefinition function_library_; member in tensorflow::grappler::__anon63eee5310111::AutoMixedPrecisionImpl
1583 return source_and_sink_ops.count(op) || function_library_.Find(op); in IsSourceOrSinkOp()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 650 function_library_(OpRegistry::Global(), graph.graph()->library()), in SymbolicShapeRefiner()
1314 CHECK_NOTNULL(function_library_.Find(function_name)); in AddFunction()
1317 MakeGrapplerFunctionItem(*function_def, function_library_, in AddFunction()
1359 function_library_.LookUp(function.name(), &node_ctx.op_data)); in AddNode()
2052 FunctionLibraryDefinition function_library_; member in tensorflow::grappler::SymbolicShapeRefiner
|