Searched defs:transpose1 (Results 1 - 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | instruction_fusion_test.cc | 61 auto transpose1 = builder.AddInstruction(HloInstruction::CreateTranspose( local
66 EXPECT_EQ(transpose1, computation->root_instruction());
|
| H A D | transpose_folding_test.cc | 107 HloInstruction* transpose1 = local
115 /*lhs=*/transpose0, /*rhs=*/transpose1, dot_dnums));
|
| H A D | algebraic_simplifier_test.cc | 1496 HloInstruction* transpose1 = local
1501 ShapeUtil::MakeShape(F32, {4, 3, 2}), transpose1, {1, 0, 2}));
1505 EXPECT_THAT(computation->root_instruction(), op::Transpose(transpose1));
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_instruction_fusion_test.cc | 170 HloInstruction* transpose1 = local
174 MakeDot(ShapeUtil::MakeShape(F32, {1, 1024}), arg0, transpose1));
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | fusion_test.cc | 559 auto transpose1 = builder.AddInstruction(HloInstruction::CreateTranspose( local
562 ShapeUtil::MakeShape(S32, {2, 2}), HloOpcode::kNegate, transpose1));
564 ->CreateFusionInstruction(/*instructions_to_fuse=*/{negate2, transpose1},
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | arithmetic_optimizer_test.cc | 791 Output transpose1 = ops::Transpose(s.WithOpName("transpose1"), inputs, perm1); local
793 ops::Transpose(s.WithOpName("transpose2"), transpose1, perm2);
852 Output transpose1 = ops::Transpose(s, inputs, ops::Const(s, {1, 0})); local
853 Output transpose2 = ops::Transpose(s, transpose1, ops::Const(s, {1, 0}));
880 Output transpose1 = ops::Transpose(s.WithOpName("transpose1"), inputs, perm); local
882 ops::Transpose(s.WithOpName("transpose2"), transpose1, perm);
|
Completed in 166 milliseconds