Searched refs:broadcast1 (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | fusion_test.cc | 505 auto broadcast1 = builder.AddInstruction(HloInstruction::CreateBroadcast( in XLA_TEST_F() local
508 ShapeUtil::MakeShape(S32, {2}), HloOpcode::kNegate, broadcast1)); in XLA_TEST_F()
510 ->CreateFusionInstruction(/*instructions_to_fuse=*/{negate2, broadcast1}, in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 371 HloInstruction* broadcast1 = builder.AddInstruction( in TEST_F() local
374 result_shape, HloOpcode::kNegate, broadcast1)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 2293 auto broadcast1 = builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F() local
2296 ShapeUtil::MakeShape(F32, {2, 3, 7, 2, 1, 3, 2}), broadcast1)); in TEST_F()
|