Searched refs:shape_inference_graph (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_XlaHostCompute.pbtxt | 37 If shape_inference_graph is empty, a list of the shapes of `outputs`.
41 name: "shape_inference_graph"
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 88 NameAttrList shape_inference_graph; in HostComputeOp() local
90 ctx, ctx->GetAttr("shape_inference_graph", &shape_inference_graph)); in HostComputeOp()
91 if (shape_inference_graph.name().empty()) { in HostComputeOp()
108 const FunctionDef* fdef = library->Find(shape_inference_graph.name()); in HostComputeOp()
111 shape_inference_graph.name(), in HostComputeOp()
114 *fdef, AttrSlice(&shape_inference_graph.attr()), in HostComputeOp()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | XlaHostCompute.pbtxt | 32 name: "shape_inference_graph"
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 1211 const FunctionDef* shape_inference_graph = in RewriteShapeInferenceGraph() local
1213 TF_RET_CHECK(shape_inference_graph); in RewriteShapeInferenceGraph()
1214 TF_RETURN_IF_ERROR(FunctionDefToBodyHelper(*shape_inference_graph, in RewriteShapeInferenceGraph()
2287 NameAttrList shape_inference_graph; in operator ()() local
2288 AddNodeAttr("shape_inference_graph", shape_inference_graph, node_def); in operator ()()
2293 NameAttrList shape_inference_graph; in operator ()() local
2294 shape_inference_graph.set_name(shape_inference_func_name); in operator ()()
2295 AddNodeAttr("shape_inference_graph", shape_inference_graph, node_def); in operator ()()
2381 auto shape_inference_graph = absl::make_unique<NameAttrList>(); in ExtractOutsideCompilationForFunction() local
2383 shape_inference_graph.get())); in ExtractOutsideCompilationForFunction()
[all …]
|
| D | encapsulate_subgraphs_pass_test.cc | 915 NameAttrList shape_inference_graph; in TEST() local
916 shape_inference_graph.set_name( in TEST()
936 {"shape_inference_graph", shape_inference_graph}, in TEST()
1755 NameAttrList shape_inference_graph; in TEST() local
1756 shape_inference_graph.set_name( in TEST()
1772 {"shape_inference_graph", shape_inference_graph}, in TEST()
1863 NameAttrList shape_inference_graph; in TEST() local
1864 shape_inference_graph.set_name( in TEST()
1884 {"shape_inference_graph", shape_inference_graph}, in TEST()
2148 NameAttrList shape_inference_graph; in TEST() local
[all …]
|
| D | extract_outside_compilation_pass_test.cc | 117 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()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 58651 name: "shape_inference_graph"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 17827 SymbolRefAttr:$shape_inference_graph,
|