Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dextract_outside_compilation_pass.cc1217 const FunctionDef* shape_inference_graph = in RewriteShapeInferenceGraph() local
1219 TF_RET_CHECK(shape_inference_graph); in RewriteShapeInferenceGraph()
1220 TF_RETURN_IF_ERROR(FunctionDefToBodyHelper(*shape_inference_graph, in RewriteShapeInferenceGraph()
2288 NameAttrList shape_inference_graph; in operator ()() local
2289 AddNodeAttr("shape_inference_graph", shape_inference_graph, node_def); in operator ()()
2294 NameAttrList shape_inference_graph; in operator ()() local
2295 shape_inference_graph.set_name(shape_inference_func_name); in operator ()()
2296 AddNodeAttr("shape_inference_graph", shape_inference_graph, node_def); in operator ()()
2382 auto shape_inference_graph = std::make_unique<NameAttrList>(); in ExtractOutsideCompilationForFunction() local
2384 shape_inference_graph.get())); in ExtractOutsideCompilationForFunction()
[all …]
H A Dencapsulate_subgraphs_pass_test.cc915 NameAttrList shape_inference_graph; in TEST() local
916 shape_inference_graph.set_name( in TEST()
938 {"shape_inference_graph", shape_inference_graph}, in TEST()
1773 NameAttrList shape_inference_graph; in TEST() local
1774 shape_inference_graph.set_name( in TEST()
1792 {"shape_inference_graph", shape_inference_graph}, in TEST()
1883 NameAttrList shape_inference_graph; in TEST() local
1884 shape_inference_graph.set_name( in TEST()
1906 {"shape_inference_graph", shape_inference_graph}, in TEST()
2174 NameAttrList shape_inference_graph; in TEST() local
[all …]
H A Dextract_outside_compilation_pass_test.cc117 NameAttrList shape_inference_graph; in TEST() local
119 "shape_inference_graph", &shape_inference_graph)); in TEST()
120 EXPECT_EQ(shape_inference_graph.name(), in TEST()
330 NameAttrList shape_inference_graph; in TEST_F() local
332 &shape_inference_graph)); in TEST_F()
333 EXPECT_EQ(shape_inference_graph.name(), ""); in TEST_F()
335 &shape_inference_graph)); in TEST_F()
336 EXPECT_EQ(shape_inference_graph.name(), ""); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
H A DXlaHostCompute.pbtxt32 name: "shape_inference_graph"
86 name: "shape_inference_graph"
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/
H A Dhost_compute_ops.cc99 NameAttrList shape_inference_graph; in HostComputeOp() local
101 ctx, ctx->GetAttr("shape_inference_graph", &shape_inference_graph)); in HostComputeOp()
102 const std::string& shape_inference_func_name = shape_inference_graph.name(); in HostComputeOp()
126 *fdef, AttrSlice(&shape_inference_graph.attr()), in HostComputeOp()
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_XlaHostCompute.pbtxt37 If shape_inference_graph is empty, a list of the shapes of `outputs`.
41 name: "shape_inference_graph"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/
H A Doptional_symbol_ref.mlir17 // CHECK: key: "shape_inference_graph"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
H A Dprepare_tpu_computation_for_tf_export.mlir26 // CHECK-SAME: shape_inference_graph = @host_func
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops.td1472 // Declares symbol reference attribute `shape_inference_graph` to be optional
1485 OptionalAttr<SymbolRefAttr>:$shape_inference_graph,
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dops.pbtxt63055 name: "shape_inference_graph"