Home
last modified time | relevance | path

Searched refs:while_shape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_invariant_code_motion_test.cc63 Shape while_shape = in TEST_F() local
69 HloInstruction::CreateParameter(0, while_shape, "param")); in TEST_F()
85 HloInstruction::CreateParameter(0, while_shape, "init_value")); in TEST_F()
87 while_shape, MakeAlwaysTrueComputation(while_shape, m.get()), while_body, in TEST_F()
105 Shape while_shape = in TEST_F() local
111 HloInstruction::CreateParameter(0, while_shape, "param")); in TEST_F()
144 HloInstruction::CreateParameter(0, while_shape, "init_value")); in TEST_F()
146 while_shape, MakeAlwaysTrueComputation(while_shape, m.get()), while_body, in TEST_F()
178 Shape while_shape = ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32}); in TEST_F() local
183 HloInstruction::CreateParameter(0, while_shape, "param")); in TEST_F()
[all …]
Dgather_expander_test.cc91 const Shape& while_shape = while_instr->shape(); in TEST() local
92 ASSERT_TRUE(while_shape.IsTuple()); in TEST()
93 ASSERT_EQ(ShapeUtil::TupleElementCount(while_shape), 4); in TEST()
97 ShapeUtil::GetTupleElementShape(while_shape, 1))); in TEST()
101 ShapeUtil::GetTupleElementShape(while_shape, 2))); in TEST()
105 ShapeUtil::GetTupleElementShape(while_shape, 3))); in TEST()
Dwhile_loop_simplifier.cc328 const auto& while_shape = while_init->shape(); in TryRemoveConstantParams() local
329 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) { in TryRemoveConstantParams()
346 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) { in TryRemoveConstantParams()
348 new_while_shape_elems.push_back(while_shape.tuple_shapes(i)); in TryRemoveConstantParams()
364 CHECK(ShapeUtil::Compatible(instr->shape(), while_shape)); in TryRemoveConstantParams()
367 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) { in TryRemoveConstantParams()
371 while_shape.tuple_shapes(i), instr, i))); in TryRemoveConstantParams()
382 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) { in TryRemoveConstantParams()
388 while_shape.tuple_shapes(i), instr, j))); in TryRemoveConstantParams()
658 Shape while_shape = while_init->shape(); in TryFlattenNestedTuples() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dparams_test.cc378 Shape while_shape = ShapeUtil::MakeTupleShape(parameter_shapes); in XLA_TEST_F() local
383 Parameter(&builder, 0, while_shape, "condition_parameter"); in XLA_TEST_F()
393 auto body_parameter = Parameter(&builder, 0, while_shape, "body_parameter"); in XLA_TEST_F()
Dwhile_test.cc1192 auto while_shape = ShapeUtil::MakeTupleShape( in XLA_TEST_F() local
1199 auto state = Parameter(&builder, 0, while_shape, "state"); in XLA_TEST_F()
1207 auto state = Parameter(&builder, 0, while_shape, "state"); in XLA_TEST_F()
1234 auto while_shape = ShapeUtil::MakeShape(S32, {}); in XLA_TEST_F() local
1239 Parameter(&builder, 0, while_shape, "state"); in XLA_TEST_F()
1247 auto indvar = Parameter(&builder, 0, while_shape, "state"); in XLA_TEST_F()