Search Options

Results per page
Sort
Preferred Languages
Advance

Results 31 - 40 of 179 for setAttrs (0.23 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/launch_to_device_attribute.cc

        if (op->getDialect() != tf_dialect) return WalkResult::advance();
    
        if (parallel_group_attr) {
          op->setAttr(TF::kParallelExecAnnotation, parallel_group_attr);
        }
        auto device_attr = op->getAttr(kDeviceAttr);
        if (!device_attr) {
          op->setAttr(kDeviceAttr, launch.getDeviceAttr());
          return WalkResult::advance();
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc

      builder.insert(metadata_op.clone());
      for (Operation& op : func.getOps()) {
        if (!IsTPUOp(&op)) continue;
        op.setAttr(TF::kReplicationInfoAttr, new_group);
      }
      TF::TPUCompilationResultOp new_result = compilation_op.clone();
      new_result->setAttr(kTpuCompilationStatus, new_group);
      builder.insert(new_result);
    }
    
    template <typename OpType>
    LogicalResult FindAndExcludeOp(func::FuncOp func,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.4K bytes
    - Viewed (0)
  3. pkg/registry/core/node/strategy.go

    		"spec.unschedulable": fmt.Sprint(node.Spec.Unschedulable),
    	}
    	return generic.MergeFieldsSets(objectMetaFieldsSet, specificFieldsSet)
    }
    
    // GetAttrs returns labels and fields of a given object for filtering purposes.
    func GetAttrs(obj runtime.Object) (labels.Set, fields.Set, error) {
    	nodeObj, ok := obj.(*api.Node)
    	if !ok {
    		return nil, nil, fmt.Errorf("not a node")
    	}
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Wed Mar 13 23:06:39 UTC 2024
    - 9K bytes
    - Viewed (0)
  4. staging/src/k8s.io/apiextensions-apiserver/pkg/registry/customresourcedefinition/strategy.go

    // WarningsOnUpdate returns warnings for the given update.
    func (statusStrategy) WarningsOnUpdate(ctx context.Context, obj, old runtime.Object) []string {
    	return nil
    }
    
    // GetAttrs returns labels and fields of a given object for filtering purposes.
    func GetAttrs(obj runtime.Object) (labels.Set, fields.Set, error) {
    	apiserver, ok := obj.(*apiextensions.CustomResourceDefinition)
    	if !ok {
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Wed Feb 28 19:06:46 UTC 2024
    - 10.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting_cleanup.cc

        if (cloned == func) continue;
        // Patch up the op attribute to point to the new function.
        for (NamedAttribute attr : op->getAttrs()) {
          auto symref = mlir::dyn_cast<FlatSymbolRefAttr>(attr.getValue());
          if (!symref) continue;
          if (symref.getValue() != func.getName()) continue;
          op->setAttr(attr.getName(),
                      FlatSymbolRefAttr::get(op->getContext(), cloned.getName()));
          break;
        }
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  6. tensorflow/cc/framework/cc_op_gen_util.cc

        strings::StrAppend(&setters, attr_comment);
        strings::StrAppend(&setters, "    TF_MUST_USE_RESULT Attrs ", attr_func_def,
                           " x) {\n");
        strings::StrAppend(&setters, "      Attrs ret = *this;\n");
        strings::StrAppend(&setters, "      ret.", api_def_attr.rename_to(),
                           "_ = x;\n");
        strings::StrAppend(&setters, "      return ret;\n    }\n\n");
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 00:57:05 UTC 2024
    - 25K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_deserialization.cc

            return func.emitError()
                   << "failed to rename StableHLO function " << func.getSymName();
          }
        }
        if (is_main_func) {
          main_func_name = func.getSymNameAttr();
        }
        func->setAttr(kFromXlaCallModuleAttrName, builder.getUnitAttr());
      }
      if (!main_func_name) {
        return stablehlo_module.emitError()
               << "StableHLO module does not have an entry function";
      }
      return main_func_name;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 23 09:05:47 UTC 2024
    - 11.1K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_asset_sinking_pass.cc

      // Name of the main function for the eventual executable needs to be set.
      main_func->setAttr(kTfSavedModelExportedNamesAttr,
                         builder.getStrArrayAttr({func_name}));
      main_func->setAttr(
          kTfSavedModelInitializerTypeAttr,
          builder.getStringAttr(kTfSavedModelInitializerRestoreType));
      module->setAttr("tf_saved_model.semantics",
                      UnitAttr::get(builder.getContext()));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/quantization/lite/tfl_to_std.cc

                                              dq.getArg());
          dq.getResult().replaceAllUsesWith(dcast);
          if (auto extra_attr = op->getAttr(mlir::quant::kVolatileOpAttrName)) {
            dcast->setAttr(mlir::quant::kVolatileOpAttrName, extra_attr);
          }
          dq.erase();
        } else if (auto q = llvm::dyn_cast<quantfork::QuantizeCastOp>(op)) {
          auto out_type = q.getResult().getType();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 02:50:01 UTC 2024
    - 3.5K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/stablehlo/cc/calibration/statistics.cc

          return;
        }
    
        const auto [min_value, max_value] = *min_max_values;
        mlir::OpBuilder builder(aggregator_op);
        aggregator_op->setAttr("min", builder.getF32FloatAttr(min_value));
        aggregator_op->setAttr("max", builder.getF32FloatAttr(max_value));
      });
      return status;
    }
    
    bool IsCalibrationRequired(mlir::ModuleOp module_op) {
      bool calibration_required = false;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 06:31:57 UTC 2024
    - 4.6K bytes
    - Viewed (0)
Back to top