| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| local_client_aot_test_helper.cc | 39 auto r0f32 = xla::ShapeUtil::MakeShape(xla::F32, {}); local
40 auto x = xla::Parameter(&builder, 0, r0f32, "x");
55 auto r0f32 = xla::ShapeUtil::MakeShape(xla::F32, {}); local
56 auto sum = CustomCall(&builder, "SumStructElements", {opaque_param}, r0f32);
81 &computation, /*argument_layouts=*/{&opaque_shape}, &r0f32};
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| map_inliner_test.cc | 42 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
46 HloInstruction::CreateParameter(0, r0f32, "x"));
48 HloInstruction::CreateParameter(1, r0f32, "y"));
79 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
83 HloInstruction::CreateParameter(0, r0f32, "x"));
112 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
118 HloInstruction::CreateParameter(1, r0f32, "x"));
120 HloInstruction::CreateParameter(0, r0f32, "y"));
150 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
153 param_builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32, "p0"))
[all...] |
| call_inliner_test.cc | 60 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
62 HloInstruction::CreateCall(r0f32, {}, inner_computation));
|
| hlo_reachability_test.cc | 101 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
108 r0f32, HloOpcode::kAdd, constant1, constant2));
110 HloInstruction::CreateUnary(r0f32, HloOpcode::kNegate, constant2));
112 HloInstruction::CreateUnary(r0f32, HloOpcode::kExp, negate));
114 HloInstruction::CreateBinary(r0f32, HloOpcode::kMultiply, add, exp));
116 HloInstruction::CreateUnary(r0f32, HloOpcode::kCopy, exp));
|
| heap_simulator_test.cc | 131 const Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
144 cond_builder.AddInstruction(HloInstruction::CreateReshape(r0f32, slice));
[all...] |
| hlo_instruction_test.cc | 329 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
336 HloInstruction::CreateParameter(0, r0f32, "x"));
340 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, param, value));
367 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
374 HloInstruction::CreateParameter(0, r0f32, "x"));
376 HloInstruction::CreateParameter(1, r0f32, "y"));
378 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, paramx, paramy));
1097 const Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
[all...] |
| algebraic_simplifier_test.cc | 58 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
61 HloInstruction::CreateParameter(0, r0f32, "param0"));
65 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, param0, zero));
403 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
406 HloInstruction::CreateParameter(0, r0f32, "param0"));
410 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, constant, param0));
424 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
427 HloInstruction::CreateParameter(0, r0f32, "param0"));
434 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, param0, constant1));
436 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, add1, constant2))
603 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
624 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
646 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
677 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
745 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
772 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
906 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
1020 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
1047 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
1077 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
1106 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
1133 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
1160 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
1183 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
1215 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
1269 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
1292 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
1316 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
1499 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
1546 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
1789 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
3506 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| cpu_fusion_test.cc | 158 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); local
161 r0f32, HloOpcode::kAdd,
163 HloInstruction::CreateParameter(0, r0f32, "x")),
165 HloInstruction::CreateParameter(1, r0f32, "y"))));
|
