- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetOptimalLayout (0.4 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
LogicalResult FusedBatchNormV2Op::UpdateDataFormat(StringRef data_format) { return ::mlir::TF::UpdateDataFormat(data_format, this); } StringRef FusedBatchNormV2Op::GetOptimalLayout(const RuntimeDevices& devices) { return ::mlir::TF::GetOptimalLayout(devices, this); } LogicalResult FusedBatchNormV3Op::FoldOperandsPermutation( ArrayRef<int64_t> permutation) {
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/transforms/layout_optimization.cc
// Get desired op data format. StringRef target_data_format = force_data_format_; if (target_data_format.empty()) { target_data_format = layout_sensitive_interface.GetOptimalLayout(devices); } // Skip ops that already use target data format. auto data_format = layout_sensitive_interface.data_format(); if (data_format == target_data_format) return;
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
// TF_LayoutSensitiveInterface: 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]> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)