Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass_test.cc915 NameAttrList shape_inference_graph; in TEST() local
1755 NameAttrList shape_inference_graph; in TEST() local
1863 NameAttrList shape_inference_graph; in TEST() local
2148 NameAttrList shape_inference_graph; in TEST() local
2298 NameAttrList shape_inference_graph; in TEST() local
2459 NameAttrList shape_inference_graph; in TEST() local
2574 NameAttrList shape_inference_graph; in TEST() local
Dextract_outside_compilation_pass.cc1211 const FunctionDef* shape_inference_graph = in RewriteShapeInferenceGraph() local
2287 NameAttrList shape_inference_graph; in operator ()() local
2293 NameAttrList shape_inference_graph; in operator ()() local
2381 auto shape_inference_graph = absl::make_unique<NameAttrList>(); in ExtractOutsideCompilationForFunction() local
2460 for (const string& shape_inference_graph : in ExtractOutsideCompilationForFunction() local
Dextract_outside_compilation_pass_test.cc117 NameAttrList shape_inference_graph; in TEST() local
330 NameAttrList shape_inference_graph; in TEST_F() local
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dhost_compute_ops.cc88 NameAttrList shape_inference_graph; in HostComputeOp() local