Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dfusion_utils_test.cc56 auto *parent_function = graph.mutable_library()->add_function(); in TEST() local
57 *parent_function = test::function::XTimesTwo(); in TEST()
62 *parent_function, *function, "fused_maps", fusion_utils::ComposeSignature, in TEST()
134 auto *parent_function = graph.mutable_library()->add_function(); in TEST() local
135 *parent_function = test::function::XTimesTwo(); in TEST()
140 *parent_function, *function, "fused_maps", fusion_utils::CombineSignature, in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dmove_compilation_to_host.cc227 auto parent_function = in HandleCompilationOps() local
231 host_function = parent_function; in HandleCompilationOps()
245 if (host_function != parent_function) in HandleCompilationOps()
H A Dmesh_propagation.cc402 auto parent_function = in AnnotateFunctionReturnValuesWithMeshInformation() local
405 parent_function.getResultAttrOfType<mlir::StringAttr>( in AnnotateFunctionReturnValuesWithMeshInformation()
411 return parent_function.emitOpError( in AnnotateFunctionReturnValuesWithMeshInformation()
418 parent_function.getResultAttrOfType<mlir::StringAttr>( in AnnotateFunctionReturnValuesWithMeshInformation()