- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 355 for setAttr (1.97 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.cc
/*disabled_checks=*/empty_array_attr); xla_call_module_op->setAttr(TF::kStablehloEntryFunctionAttrName, SymbolRefAttr::get(stablehlo_func_op)); // Set jax.uses_shape_polymorphism=true to enable shape refinement at runtime. // This is needed for native serialization version >= 8. xla_call_module_op->setAttr( kStablehloModuleAttrsAttrName,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_host_computation_expansion.cc
should_expand_op_to_host_computation = false; continue; } } if (should_expand_op_to_host_computation) head_outside_compiled_op->setAttr(kXlaOutsideCompilationAttr, builder->getStringAttr("")); } } #define GEN_PASS_DEF_TPUHOSTCOMPUTATIONEXPANSIONPASS
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/flatbuffer_import.cc
// use it in exports or if there are breaking changes module->setAttr("tfl.schema_version", builder.getI32IntegerAttr(model->version)); if (!model->description.empty()) { module->setAttr("tfl.description", builder.getStringAttr(model->description)); } if (!metadata_attrs.empty()) { module->setAttr("tfl.metadata", builder.getDictionaryAttr(metadata_attrs)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/quantization_context.cc
input_specs.push_back(TypeAttr::get(requantize.params)); } else { input_specs.push_back(TypeAttr::get(state.params)); } } op->setAttr("input_specs", ArrayAttr::get(context, input_specs)); llvm::SmallVector<Attribute, 4> output_specs; auto original_output_specs = op.getOutputSpecs().getValue();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 08 01:38:03 UTC 2024 - 13.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/unfreeze_constants.cc
/*arg_types=*/{}, /*arg_locs=*/{}); init_func->setAttr(kTfSavedModelExportedNamesAttr, builder.getStrArrayAttr( {"tf_saved_model.session_initializer_restore_op"})); init_func->setAttr( kTfSavedModelInitializerTypeAttr, builder.getStringAttr(kTfSavedModelInitializerRestoreType));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 16 15:04:53 UTC 2023 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/colocate_tpu_copy_with_dynamic_shape.cc
state = solver.lookupState<Device>(result); if (state) break; } if (!state || !state->hasDevice()) { return WalkResult::advance(); } op->setAttr(kDevice, state->getDevice()); return WalkResult::advance(); }); } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateColocateTPUCopyWithDynamicShapePass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 23 00:30:27 UTC 2023 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/device_util.cc
devices.push_back(builder.getNamedAttr(name, metadata)); } } op->setAttr(kDevicesAttr, builder.getDictionaryAttr(devices)); } mlir::LogicalResult GetDevicesFromOp(mlir::Operation* op, mlir::TF::RuntimeDevices* devices) { auto devices_attr = op->getAttr(kDevicesAttr); if (!devices_attr) return mlir::success();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_annotate_dynamic_shape_inputs.cc
std::get<1>(result).getDefiningOp())) { dynamic_shape_arg_index.push_back(cluster_func_operand.index()); } } } cluster_func_op->setAttr(TF::kDynamicArgIndexAttr, builder.getI32ArrayAttr(dynamic_shape_arg_index)); FlatSymbolRefAttr func_attr = cluster_func_op.getFuncAttr(); func::FuncOp func =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/target_annotation.cc
OpBuilder* builder) { // TODO(karimnosseir): Maybe set device capabilities to allow us to have // more flexbility when raise the subgraphs. auto default_target = builder->getStringAttr(annotation); op->setAttr(attribute, default_target); } void TargetAnnotationPass::SetTargetAnnotation( Operation* op, llvm::ArrayRef<std::string> device_specs, OpBuilder* builder) { if (op->hasAttr(kSkipTargetAnnotation)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 19 19:32:06 UTC 2023 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/gpu_fusion.cc
if (!add_op || add_op.getZ().hasOneUse()) { // We fuse the Relu only if the add has a single use, otherwise we only // fuse the add itself. op->setAttr("activation_mode", rewriter.getStringAttr("Relu")); rewriter.replaceOp(relu_op, op->getResult(0)); } if (add_op) { rewriter.replaceOp(add_op, op->getResult(0)); } return success();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 5.2K bytes - Viewed (0)