Searched refs:r2f32_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | optimize_input_output_buffer_alias_test.cc | 36 r2f32_ = ShapeUtil::MakeShape(F32, {4, 5}); in OptimizeInputOutputBufferAliasTest()
72 Shape r2f32_; member in xla::OptimizeInputOutputBufferAliasTest
79 Shape input = ShapeUtil::MakeTupleShape({r1f32_, r2f32_}); in TEST_F()
88 Shape input = ShapeUtil::MakeTupleShape({r1f32_, r2f32_, r3f32_, r4f32_}); in TEST_F()
89 Shape output = ShapeUtil::MakeTupleShape({r1f32_, r2f32_, r3f32_, r4f32_}); in TEST_F()
103 Shape input = ShapeUtil::MakeTupleShape({r1f32_, r1f32_, r2f32_, r2f32_}); in TEST_F()
104 Shape output = ShapeUtil::MakeTupleShape({r1f32_, r2f32_, r3f32_, r4f32_}); in TEST_F()
117 Shape input = ShapeUtil::MakeTupleShape({r1f32_, r2f32_, r3f32_, r4f32_}); in TEST_F()
118 Shape output = ShapeUtil::MakeTupleShape({r4f32_, r3f32_, r2f32_, r1f32_}); in TEST_F()
132 {ShapeUtil::MakeTupleShape({r1f32_}), r2f32_, r3f32_, r4f32_}); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | custom_call_test.cc | 78 Shape r2f32_ = ShapeUtil::MakeShape(F32, {2, 2}); member in xla::__anon047ef51e0211::CustomCallTest
146 b.AddInstruction(HloInstruction::CreateParameter(0, r2f32_, "p")); in XLA_TEST_F()
148 HloInstruction::CreateCustomCall(r2f32_, {input}, "Add1ToValues")); in XLA_TEST_F()
171 b.AddInstruction(HloInstruction::CreateParameter(0, r2f32_, "p")); in XLA_TEST_F()
|