Home
last modified time | relevance | path

Searched refs:shape3 (Results 1 – 6 of 6) sorted by relevance

/external/llvm-project/mlir/test/Dialect/Standard/
Dops.mlir71 %shape2: memref<2xi32>, %shape3: memref<?xi32>) -> memref<*xf32> {
76 %new_unranked = memref_reshape %dyn_mat(%shape3)
/external/tensorflow/tensorflow/compiler/xla/service/
Dreshape_mover_test.cc495 auto shape3 = ShapeUtil::MakeShape(F32, {8, 7}); in TEST_F() local
510 builder.AddInstruction(HloInstruction::CreateReshape(shape3, param2)); in TEST_F()
512 builder.AddInstruction(HloInstruction::CreateReshape(shape3, add0)); in TEST_F()
513 builder.AddInstruction(HloInstruction::CreateBinary(shape3, HloOpcode::kAdd, in TEST_F()
Dmemory_space_assignment_test.cc961 Shape shape3 = ShapeUtil::MakeShape(F32, {1, 6}); in TEST_P() local
982 builder.AddInstruction(HloInstruction::CreateBitcast(shape3, p1)); in TEST_P()
2525 Shape shape3 = ShapeUtil::MakeShape(F32, {2, 3}); in TEST_P() local
2528 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape3, shape4}); in TEST_P()
2534 HloInstruction::CreateGetTupleElement(shape3, p0, 0)); in TEST_P()
2562 HloInstruction::CreatePad(shape3, add2, padding_value, padding_config)); in TEST_P()
2566 HloInstruction::CreateBinary(shape3, HloOpcode::kAdd, padded_add2, p3)); in TEST_P()
Dalgebraic_simplifier_test.cc5987 auto shape3 = ShapeUtil::MakeShape(F32, {2, 3, 2}); in TEST_F() local
5997 .WithShapeCompatibleTo(&shape3))))); in TEST_F()
6017 auto shape3 = ShapeUtil::MakeShape(F32, {2, 2, 3}); in TEST_F() local
6023 .WithShapeCompatibleTo(&shape3))))); in TEST_F()
6042 auto shape3 = ShapeUtil::MakeShape(F32, {2, 2, 3}); in TEST_F() local
6048 .WithShapeCompatibleTo(&shape3))))); in TEST_F()
6095 auto shape3 = ShapeUtil::MakeShape(F32, {2, 2, 3, 2}); in TEST_F() local
6103 .WithShapeCompatibleTo(&shape3))))); in TEST_F()
6216 auto shape3 = ShapeUtil::MakeShape(F32, {1, 2, 3, 4}); in TEST_F() local
6224 .WithShapeCompatibleTo(&shape3))))); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dconstant-fold.mlir192 %shape3 = constant dense<0> : tensor<3xi32>
198 …%offset:4 = "tf.ConcatOffset"(%concat_dim, %shape0, %shape1, %shape2, %shape3) : (tensor<i32>, ten…
Dtf-ops.mlir346 %shape3 = constant dense<[-1, 100]> : tensor<2xi32>
347 %r4 = "tf.Reshape"(%arg2, %shape3) : (tensor<10000xf32>, tensor<2xi32>) -> tensor<100x100xf32>