| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_memory_scheduler_test.cc | 153 const Shape r1f32 = ShapeUtil::MakeShape(xla::F32, {6}); local
160 HloInstruction::CreateUnary(r1f32, HloOpcode::kAbs, lit));
163 HloInstruction::CreateUnary(r1f32, HloOpcode::kAbs, abs_const));
167 HloInstruction::CreateGetTupleElement(r1f32, tuple, 0));
170 HloInstruction::CreateUnary(r1f32, HloOpcode::kAbs, abs_const));
172 builder.AddInstruction(HloInstruction::CreateBinary(r1f32, HloOpcode::kAdd,
197 const Shape r1f32 = ShapeUtil::MakeShape(xla::F32, {5}); local
208 HloInstruction::CreateBinary(r1f32, HloOpcode::kAdd, c1, c2));
210 HloInstruction::CreateBinary(r1f32, HloOpcode::kMultiply, add, c3));
214 HloInstruction::CreateGetTupleElement(r1f32, tuple, 0))
247 const Shape r1f32 = ShapeUtil::MakeShape(F32, {4}); local
[all...] |
| algebraic_simplifier_test.cc | 389 Shape r1f32 = ShapeUtil::MakeShape(F32, {5}); local
390 builder.AddInstruction(HloInstruction::CreateReduce(r1f32, reduce0, zero,
805 Shape r1f32 = ShapeUtil::MakeShape(F32, {7}); local
837 Shape r1f32 = ShapeUtil::MakeShape(F32, {3}); local
859 Shape r1f32 = ShapeUtil::MakeShape(F32, {7}); local
1240 Shape r1f32 = ShapeUtil::MakeShape(F32, {42}); local
1654 Shape r1f32 = ShapeUtil::MakeShape(F32, {100}); local
1676 Shape r1f32 = ShapeUtil::MakeShape(F32, {kParamLength}); local
1760 Shape r1f32 = ShapeUtil::MakeShape(F32, {kParamLength}); local
1788 Shape r1f32 = ShapeUtil::MakeShape(F32, {100}); local
3761 Shape r1f32 = ShapeUtil::MakeShape(F32, {1}); local
[all...] |
| hlo_instruction_test.cc | 1002 const Shape r1f32 = ShapeUtil::MakeShape(F32, {5}); local
1037 const Shape r1f32 = ShapeUtil::MakeShape(F32, {5}); local
1098 const Shape r1f32 = ShapeUtil::MakeShape(F32, {5}); local
[all...] |
| heap_simulator_test.cc | 132 const Shape r1f32 = ShapeUtil::MakeShape(F32, {4}); local
139 HloInstruction::CreateParameter(0, r1f32, "cond_param"));
156 HloInstruction::CreateParameter(0, r1f32, "body_param"));
162 r1f32, HloOpcode::kSubtract, body_param, one_vector));
173 r1f32, cond_computation, body_computation, while_init));
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| concat_test.cc | 412 auto r1f32 = xla::ShapeUtil::MakeShape(xla::F32, {1}); local
413 auto x = Parameter(&builder, 0, r1f32, "x");
427 auto r1f32 = xla::ShapeUtil::MakeShape(xla::F32, {1}); local
428 auto x = Parameter(&builder, 0, r1f32, "x");
|
