Searched refs:get_dimension_size (Results 1 – 8 of 8) sorted by relevance
160 auto get_dimension_size = GetDimensionSize(add, 0); in TEST_F() local161 EXPECT_TRUE(IsConstant(get_dimension_size, &b)); in TEST_F()164 client, get_dimension_size, &b)); in TEST_F()175 auto get_dimension_size = GetDimensionSize(add, 0); in TEST_F() local177 auto add_2 = Add(get_dimension_size, get_dimension_size_2); in TEST_F()
203 Status HandleGetDimensionSize(HloInstruction* get_dimension_size) override;
413 HloInstruction* get_dimension_size) { in HandleGetDimensionSize() argument414 HloInstruction* operand = get_dimension_size->mutable_operand(0); in HandleGetDimensionSize()415 int64 dim = get_dimension_size->dimension(); in HandleGetDimensionSize()424 evaluated_[get_dimension_size] = in HandleGetDimensionSize()429 const Shape& shape = get_dimension_size->operand(0)->shape(); in HandleGetDimensionSize()432 static_cast<int32>(shape.dimensions(get_dimension_size->dimension()))); in HandleGetDimensionSize()433 evaluated_[get_dimension_size] = std::move(output); in HandleGetDimensionSize()
1276 func @get_dimension_size(%I: tensor<1x128x512xf32>) -> tensor<i32> {1278 …%size = "mhlo.get_dimension_size"(%I) {dimension = 3 : i64} : (tensor<1x128x512xf32>) -> tensor<i3…1284 func @get_dimension_size(%I: tensor<1x128x512xf32>) -> tensor<i32> {1285 …%size = "mhlo.get_dimension_size"(%I) {dimension = 2 : i64} : (tensor<1x128x512xf32>) -> tensor<i3…
968 …%size = "mhlo.get_dimension_size"(%I) {dimension = 2 : i64} : (tensor<1x128x512xf32>) -> tensor<i3…
1028 def HLO_GetDimensionSizeOp: HLO_Op<"get_dimension_size", [NoSideEffect]>,
502 %1 = "mhlo.get_dimension_size"(%0) {dimension = 1 : i64} : (tensor<4x2xf32>) -> tensor<i32>
436 …// CHECK-NEXT: "mhlo.get_dimension_size"([[ARG]]) {dimension = 1 : i64} : (tensor<4x2xf32>) -> te…