Home
last modified time | relevance | path

Searched refs:dynamic_reshape (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dhlo-transform-unranked.mlir11 %flat_a = "mhlo.dynamic_reshape"(%a, %flat_shape)
18 %b = "mhlo.dynamic_reshape"(%flat_b, %shape)
33 …// CHECK-NEXT: %[[FLAT_A:.*]] = "mhlo.dynamic_reshape"(%[[A]], %[[FLAT_SHAPE]]) : (tensor<*xf32>, …
35 …// CHECK-NEXT: %[[B:.*]] = "mhlo.dynamic_reshape"(%[[FLAT_B]], %[[SHAPE]]) : (tensor<?xf32>, tenso…
75 …// CHECK: %[[FLAT_A:.*]] = "mhlo.dynamic_reshape"(%[[A]], %[[FLAT_SHAPE]]) : (tensor<*xf32>, tenso…
76 …// CHECK: %[[FLAT_B:.*]] = "mhlo.dynamic_reshape"(%[[B]], %[[FLAT_SHAPE]]) : (tensor<*xf32>, tenso…
78 …// CHECK: %[[RESULT:.*]] = "mhlo.dynamic_reshape"(%[[FLAT_RESULT]], %[[SHAPE]]) : (tensor<?xf32>, …
92 …// CHECK: %[[FLAT_A:.*]] = "mhlo.dynamic_reshape"(%[[A]], %[[FLAT_SHAPE]]) : (tensor<*xf32>, tenso…
94 …// CHECK: %[[B:.*]] = "mhlo.dynamic_reshape"(%[[FLAT_B]], %[[SHAPE]]) : (tensor<?xf32>, tensor<?xi…
117 // CHECK-NEXT: %[[RESHAPED:.*]] = "mhlo.dynamic_reshape"(%[[ARG_1]], %[[SIZE_TENSOR]]) : …
[all …]
Dhlo-legalize-to-lhlo-unranked.mlir15 %reshaped = "mhlo.dynamic_reshape"(%operand, %shape)
28 %reshaped = "mhlo.dynamic_reshape"(%operand, %shape)
Dcanonicalize.mlir583 %0 = "mhlo.dynamic_reshape"(%arg0, %shape) : (tensor<4xf32>, tensor<2xindex>) -> tensor<4x1xf32>
592 %0 = "mhlo.dynamic_reshape"(%arg0, %shape) : (tensor<*xf32>, tensor<2xindex>) -> tensor<?x?xf32>
607 %4 = "mhlo.dynamic_reshape"(%0, %3)
617 %0 = "mhlo.dynamic_reshape"(%arg0, %shape) : (tensor<?xf16>, tensor<?xindex>) -> tensor<*xf16>
621 %4 = "mhlo.dynamic_reshape"(%0, %3) : (tensor<*xf16>, tensor<1xindex>) -> tensor<?xf16>
1561 %0 = "mhlo.dynamic_reshape"(%arg, %shape)
1568 // CHECK: [[RESHAPE:%.*]] = "mhlo.dynamic_reshape"
1574 %0 = "mhlo.dynamic_reshape"(%arg, %shape)
1581 // CHECK: mhlo.dynamic_reshape
1587 %0 = "mhlo.dynamic_reshape"(%arg, %shape)
[all …]
Dops.mlir1241 %0 = "mhlo.dynamic_reshape"(%arg0, %shape) : (tensor<?xf32>, tensor<2xindex>) -> tensor<?x?xf32>
1249 %0 = "mhlo.dynamic_reshape"(%arg0, %shape) : (tensor<?xf32>, tensor<2xindex>) -> tensor<?xf32>
/external/tensorflow/tensorflow/compiler/xla/service/
Ddfs_hlo_visitor_with_default.h201 Status HandleDynamicReshape(HloInstructionPtr dynamic_reshape) override { in HandleDynamicReshape() argument
202 return DefaultAction(dynamic_reshape); in HandleDynamicReshape()
Ddynamic_dimension_inference_test.cc1268 auto dynamic_reshape = in TEST_F() local
1276 EXPECT_EQ(inference_->GetDynamicSize(dynamic_reshape, {}, 0), nullptr); in TEST_F()
1277 EXPECT_EQ(inference_->GetDynamicSize(dynamic_reshape, {}, 1), dynamic_size); in TEST_F()
1295 auto dynamic_reshape = builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F() local
1301 EXPECT_EQ(inference_->GetDynamicSize(dynamic_reshape, {}, 0), six); in TEST_F()
1302 EXPECT_EQ(inference_->GetDynamicSize(dynamic_reshape, {}, 1), nullptr); in TEST_F()
1303 EXPECT_EQ(inference_->GetDynamicSize(dynamic_reshape, {}, 2), one); in TEST_F()
Dhlo_verifier.cc715 Status ShapeVerifier::HandleDynamicReshape(HloInstruction* dynamic_reshape) { in HandleDynamicReshape() argument
717 const Shape& operand_shape = dynamic_reshape->operand(0)->shape(); in HandleDynamicReshape()
718 TF_RET_CHECK(SameElementType(dynamic_reshape->shape(), operand_shape)); in HandleDynamicReshape()
719 TF_RET_CHECK(ShapeUtil::ElementsIn(dynamic_reshape->shape()) == in HandleDynamicReshape()
721 TF_RET_CHECK(dynamic_reshape->shape().rank() + 1 == in HandleDynamicReshape()
722 dynamic_reshape->operand_count()); in HandleDynamicReshape()
723 for (int64 i = 1; i < dynamic_reshape->operand_count(); ++i) { in HandleDynamicReshape()
724 TF_RET_CHECK(dynamic_reshape->operand(i)->shape().element_type() == S32); in HandleDynamicReshape()
Dhlo_verifier.h81 Status HandleDynamicReshape(HloInstruction* dynamic_reshape) override;
Ddynamic_dimension_inference.cc806 HloDynamicReshapeInstruction* dynamic_reshape = in HandleDynamicReshape() local
810 parent_->SetDynamicSize(hlo, {}, i, dynamic_reshape->dim_sizes(i)); in HandleDynamicReshape()
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops.td1066 def HLO_DynamicReshapeOp: HLO_Op<"dynamic_reshape", [NoSideEffect]> {
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf.mlir2520 // CHECK: "mhlo.dynamic_reshape"
2565 // CHECK-DAG: [[RESHAPE:%.+]] = "mhlo.dynamic_reshape"(%arg0, [[TOEXTENTS]])
2586 // CHECK-DAG: [[RESHAPE:%.+]] = "mhlo.dynamic_reshape"(%arg0, [[TOEXTENTS]])