Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc2393 auto shape_inference_fdef = absl::make_unique<FunctionDef>(*xla_fdef); in ExtractOutsideCompilationForFunction() local
2394 shape_inference_fdef->mutable_signature()->set_name( in ExtractOutsideCompilationForFunction()
2398 shape_inference_graph->name(), *shape_inference_fdef)); in ExtractOutsideCompilationForFunction()
2400 TF_RETURN_IF_ERROR(fld->AddFunctionDef(*shape_inference_fdef)); in ExtractOutsideCompilationForFunction()