Searched refs:broadcast1 (Results 1 – 4 of 4) sorted by relevance
| /external/llvm-project/mlir/test/Dialect/Tosa/ |
| D | broadcast.mlir | 12 // CHECK-LABEL: broadcast1
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | cpu_gpu_fusion_test.cc | 504 auto broadcast1 = builder.AddInstruction(HloInstruction::CreateBroadcast( in XLA_TEST_F() local
507 ShapeUtil::MakeShape(S32, {2}), HloOpcode::kNegate, broadcast1)); in XLA_TEST_F()
509 ->CreateFusionInstruction(/*instructions_to_fuse=*/{negate2, broadcast1}, in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 373 HloInstruction* broadcast1 = builder.AddInstruction( in TEST_F() local
376 result_shape, HloOpcode::kNegate, broadcast1)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 2928 auto broadcast1 = builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F() local
2931 ShapeUtil::MakeShape(F32, {2, 3, 7, 2, 1, 3, 2}), broadcast1)); in TEST_F()
|