- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for UpdateDataFormat (0.32 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.h
DenseIntElementsAttr perm1); // Default implementation of `LayoutSensitiveInterface::UpdateDataFormat` for // layout sensitive operations that do not have any additional layout dependent // attributes besides `data_format` string. template <typename Op> LogicalResult UpdateDataFormat(StringRef data_format, Op *op) { auto perm = GetDataFormatPermutation(op->getDataFormat(), data_format);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
"found " << feature_dim << " and " << bias_len << ", respectively"; } return success(); } LogicalResult BiasAddOp::UpdateDataFormat(StringRef data_format) { return ::mlir::TF::UpdateDataFormat(data_format, this); } StringRef BiasAddOp::GetOptimalLayout(const RuntimeDevices& devices) { // Keep current data format if no GPUs are available or if explicit placement
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_op_interfaces.td
>, InterfaceMethod< [{Updates operation attributes and operands to account for the updated data format. If data format is not supported, must return failure.}], "LogicalResult", "UpdateDataFormat", (ins "StringRef":$data_format) >, ]; let verify = [{ return VerifyLayoutSensitiveInterface($_op); }]; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 30 19:07:07 UTC 2022 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc
return DenseIntElementsAttr::get(perm_ty, permutation); }; // Change operation data format. if (failed(layout_sensitive_interface.UpdateDataFormat(target_data_format))) return; // Permute arguments into the target data format. builder.setInsertionPoint(op); auto arg_perm = builder.create<ConstOp>(loc, perm_attr(args_permutation));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
SmallVector<unsigned, 4> GetLayoutDependentArgs() { return {0}; } SmallVector<unsigned, 4> GetLayoutDependentResults() { return {0}; } StringRef GetOptimalLayout(const RuntimeDevices& devices); LogicalResult UpdateDataFormat(StringRef data_format); }]; let hasVerifier = 1; } def TF_BiasAddGradOp : TF_Op<"BiasAddGrad", [Pure]> { let summary = [{ The backward operation for "BiasAdd" on the "bias" tensor.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)