- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for getI64ArrayAttr (0.39 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/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)