Home
last modified time | relevance | path

Searched defs:shape_inference_graph (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dencapsulate_subgraphs_pass_test.cc915 NameAttrList shape_inference_graph; in TEST() local
1773 NameAttrList shape_inference_graph; in TEST() local
1883 NameAttrList shape_inference_graph; in TEST() local
2174 NameAttrList shape_inference_graph; in TEST() local
2328 NameAttrList shape_inference_graph; in TEST() local
2495 NameAttrList shape_inference_graph; in TEST() local
2612 NameAttrList shape_inference_graph; in TEST() local
H A Dextract_outside_compilation_pass.cc1217 const FunctionDef* shape_inference_graph = in RewriteShapeInferenceGraph() local
2288 NameAttrList shape_inference_graph; in operator ()() local
2294 NameAttrList shape_inference_graph; in operator ()() local
2382 auto shape_inference_graph = std::make_unique<NameAttrList>(); in ExtractOutsideCompilationForFunction() local
2461 for (const string& shape_inference_graph : in ExtractOutsideCompilationForFunction() local
H A Dextract_outside_compilation_pass_test.cc117 NameAttrList shape_inference_graph; in TEST() local
330 NameAttrList shape_inference_graph; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/
H A Dhost_compute_ops.cc99 NameAttrList shape_inference_graph; in HostComputeOp() local