- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for getI64ArrayAttr (0.2 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_partitioned_op_conversion.cc
op.getLoc(), op.getType(), op.getOperands(), builder.getI64ArrayAttr(partition_dims), builder.getBoolAttr(false), op.get_XlaShardingAttr()); op->replaceAllUsesWith(pi); } else { auto po = builder.create<TF::TPUPartitionedOutputV2Op>( op.getLoc(), op.getResultTypes(), op.getOperand(), builder.getI64ArrayAttr(partition_dims), op.get_XlaShardingAttr()); op->replaceAllUsesWith(po); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
conv_op.getLoc(), conv_output_type, sliced_lhs, reshaped_filter, rewriter.getI64ArrayAttr(strides), /*padding=*/rewriter.getStringAttr(padding), /*explicit_paddings=*/rewriter.getI64ArrayAttr(new_padding), /*data_format=*/rewriter.getStringAttr("NHWC"), /*dilations=*/rewriter.getI64ArrayAttr(dilation)); } else if (num_spatial_dims == 3) { output = rewriter.create<TF::Conv3DOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_uniform_attribute_utils.cc
const int stride_w = mlir::cast<IntegerAttr>(stride[2]).getInt(); return rewriter.getI64ArrayAttr({stride_h, stride_w}); } Attribute GetLhsDilationValue(PatternRewriter& rewriter, llvm::StringMap<Attribute>& identifier_to_attr) { return rewriter.getI64ArrayAttr({1, 1}); } Attribute GetRhsDilationValue(PatternRewriter& rewriter,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.td
"CreateScalarConstValue<"# type #">($_builder, $_loc, "# value #")">; // Creates an I64 array attribute with given values. class CreateI64ArrayAttr<string values> : NativeCodeCall< "$_builder.getI64ArrayAttr("# values #")">; // Creates a string attribute with given values. class CreateStringAttr<string values> : NativeCodeCall< "$_builder.getStringAttr(\""# values #"\")">;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 08 04:55:44 UTC 2024 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_merge_variables_with_execute.cc
llvm::ArrayRef<NamedAttribute>{ builder->getNamedAttr( "device_var_reads_indices", builder->getI64ArrayAttr(device_var_reads_indices)), builder->getNamedAttr( "device_var_updates_indices", builder->getI64ArrayAttr(device_var_updates_indices))}); // Wrap in launch for device assignment. auto merged_execute_launch = builder->create<tf_device::LaunchOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 17:52:11 UTC 2024 - 27K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/dilated_conv.h
dilation_w_factor = 1; } } if (dilation_h_factor == -1 || dilation_w_factor == -1) { return {}; } return rewriter.getI64ArrayAttr({1, dilation_h_factor, dilation_w_factor, 1}); } } // namespace TFL } // namespace mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_operator.cc
} static mlir::Attribute BuildI64ArrayAttr(std::vector<int32_t> value, mlir::Builder builder) { std::vector<int64_t> typecast(value.begin(), value.end()); return builder.getI64ArrayAttr(typecast); } static mlir::Attribute BuildVhloBooleanV1Attr(bool value, mlir::Builder builder) { return mlir::vhlo::BooleanV1Attr::get(builder.getContext(), value);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 38K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
} auto in_manual = builder.create<mlir::TF::XlaSpmdFullToShardShapeOp>( loc, shard_type, in, common_split_sharding, /*dim=*/-1, /*unspecified_dims=*/builder.getI64ArrayAttr({})); manual_inputs.push_back(in_manual); } // Create the _XlaHostComputeMlirOp auto host_compute = builder.create<mlir::TF::_XlaHostComputeMlirOp>( loc, shard_output_types, manual_inputs,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
replicated_inputs, packed_inputs, result_types); if (!mirrored_variable_indices.empty()) replicate_op->setAttr(kMirroredVariableIndicesAttr, builder.getI64ArrayAttr(mirrored_variable_indices)); // Replace replicated cluster results with replicate op results. uint64_t offset = 0; for (auto [idx, result] : llvm::enumerate(cluster.getResults())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util_test.cc
} TEST(TPURewriteDeviceUtilTest, TestGetDeviceCoordinates) { mlir::MLIRContext context; mlir::Builder builder(&context); auto device_assignment_attr = builder.getI64ArrayAttr({1, 2, 3}); auto status_or_device_coodinates = GetDeviceCoordinates(device_assignment_attr); ASSERT_TRUE(status_or_device_coodinates.ok()); auto device_coordinates = status_or_device_coodinates.value();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 46.8K bytes - Viewed (0)