Searched refs:static_shaped_slice (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | increase_dynamism_for_auto_jit_pass_test.cc | 140 Node* static_shaped_slice = testing::FindNodeByName( in TEST() local
142 ASSERT_NE(static_shaped_slice, nullptr); in TEST()
143 EXPECT_THAT(static_shaped_slice, m_dynamic_slice); in TEST()
160 Node* static_shaped_slice = testing::FindNodeByName( in TEST() local
162 EXPECT_NE(static_shaped_slice, nullptr); in TEST()
182 Node* static_shaped_slice = testing::FindNodeByName( in TEST() local
184 ASSERT_NE(static_shaped_slice, nullptr); in TEST()
185 EXPECT_THAT(static_shaped_slice, in TEST()
290 Node* static_shaped_slice = testing::FindNodeByName( in TEST() local
292 ASSERT_NE(static_shaped_slice, nullptr); in TEST()
[all …]
|
| D | increase_dynamism_for_auto_jit_pass.cc | 282 Node* static_shaped_slice) { in ReplaceTensorFlowSliceWithStaticShapedSlice() argument
293 g->AddEdge(static_shaped_slice, src_output, dst, dst_input); in ReplaceTensorFlowSliceWithStaticShapedSlice()
298 g->AddControlEdge(e->src(), static_shaped_slice); in ReplaceTensorFlowSliceWithStaticShapedSlice()
309 Node* static_shaped_slice; in RewriteSlice() local
311 g, slice, slice_inputs, cluster_name, &static_shaped_slice)); in RewriteSlice()
312 ReplaceTensorFlowSliceWithStaticShapedSlice(g, slice, static_shaped_slice); in RewriteSlice()
|