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.cc100 const Shape& while_shape = while_instr->shape(); in TEST_F() local
101 ASSERT_TRUE(while_shape.IsTuple()); in TEST_F()
102 ASSERT_EQ(ShapeUtil::TupleElementCount(while_shape), 4); in TEST_F()
106 ShapeUtil::GetTupleElementShape(while_shape, 1))); in TEST_F()
110 ShapeUtil::GetTupleElementShape(while_shape, 2))); in TEST_F()
114 ShapeUtil::GetTupleElementShape(while_shape, 3))); in TEST_F()
Dwhile_loop_simplifier.cc421 auto& while_shape = while_init->shape(); in TryRemoveRepeatedWhileTupleIndices() local
443 i < while_shape.tuple_shapes_size(); ++i) { in TryRemoveRepeatedWhileTupleIndices()
509 const auto& while_shape = while_init->shape(); in TryRemoveConstantParams() local
510 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) { in TryRemoveConstantParams()
527 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) { in TryRemoveConstantParams()
529 new_while_shape_elems.push_back(while_shape.tuple_shapes(i)); in TryRemoveConstantParams()
545 CHECK(ShapeUtil::Compatible(instr->shape(), while_shape)); in TryRemoveConstantParams()
548 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) { in TryRemoveConstantParams()
552 while_shape.tuple_shapes(i), instr, i))); in TryRemoveConstantParams()
563 for (int64 i = 0; i < while_shape.tuple_shapes_size(); ++i) { in TryRemoveConstantParams()
[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()