- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for getI32TensorAttr (0.5 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_targets_test.cc
builder_.getUnknownLoc(), builder_.getI32TensorAttr({42})); ConversionTarget target = GetDefaultLegalConversionTargets(context_, /*legalize_chlo=*/true); EXPECT_TRUE(target.isIllegal(const_int)); } TEST_F(XlaLegalizeTargetsTest, DontAllowCHLODialect) { auto const_int = builder_.create<chlo::ConstantOp>( builder_.getUnknownLoc(), builder_.getI32TensorAttr({42})); ConversionTarget target =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 05:56:39 UTC 2023 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/reduce.h
auto dim_type = RankedTensorType::get({1}, rewriter.getI32Type()); auto reduction_indices = rewriter.create<arith::ConstantOp>( reduce_op.getLoc(), dim_type, rewriter.getI32TensorAttr({static_cast<int32_t>(axis)})); // Generate a Max and an ArgMax of as the mhlo op returns both while in TF // we have separate ops for them. If only one of them is used then the other
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_jax_random.cc
auto result_shape = result_ty.getShape(); for (auto element : result_shape) { result_shape_i32.push_back(static_cast<int32_t>(element)); } auto result_shape_attr = builder.getI32TensorAttr(result_shape_i32); Value result_shape_tensor = builder.create<stablehlo::ConstantOp>(result_shape_attr); auto custom_code = IsJaxRandomUniform(func) ? "RandomUniform" : "RandomStandardNormal";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0)