- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 986 for Auto (0.11 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/strip_saved_module_metadata.cc
}; bool ShouldStripAttr(NamedAttribute &namedAttr) { auto name = namedAttr.getName().strref(); return name.starts_with("tf_saved_model."); } void StripModule(Operation *module) { auto stripAttrs = llvm::to_vector<4>(llvm::make_filter_range( module->getAttrs(), [](NamedAttribute namedAttr) { return ShouldStripAttr(namedAttr); })); for (auto namedAttr : stripAttrs) { module->removeAttr(namedAttr.getName());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/mark_initialized_variables.cc
// Get resources from Session. auto resource_tensors_or = GetResourcesFromSession(var_ops, session); if (!resource_tensors_or.ok()) return function->emitError(resource_tensors_or.status().message().data()); MLIRContext* context = function.getContext(); for (auto var_and_tensor : llvm::zip(var_ops, resource_tensors_or.value())) { auto& var_op = std::get<0>(var_and_tensor); auto& resource_tensor = std::get<1>(var_and_tensor);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 13 19:14:56 UTC 2023 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/modify_io_nodes.cc
Value new_arg = arg; Location loc = func.getLoc(); if (arg.hasOneUse() && llvm::isa<QuantizeOp>(*arg.user_begin())) { auto quantize_op = llvm::cast<QuantizeOp>(*arg.user_begin()); auto quantize_output = quantize_op.getOutput(); auto current_type = quant::QuantizedType::getQuantizedElementType( quantize_output.getType()) .getStorageType();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
platforms/jvm/jvm-services/src/test/groovy/org/gradle/jvm/toolchain/internal/AutoInstalledInstallationSupplierTest.groovy
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Thu Apr 11 23:01:05 UTC 2024 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/sparsity/sparsify_model_test.cc
auto output_fbm = tflite::FlatBufferModel::BuildFromBuffer( reinterpret_cast<const char*>(output_builder.GetCurrentBufferPointer()), output_builder.GetSize()); tflite::ModelT output_model; output_fbm->GetModel()->UnPackTo(&output_model); // Extract output metadata std::map<std::string, std::string> output_metadata; for (const auto& metadata : output_model.metadata) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:16:40 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/extract_tpu_copy_with_dynamic_shape_op.cc
Operation* GetOpOfValue(Value value) { if (auto block_arg = mlir::dyn_cast<BlockArgument>(value)) return block_arg.getOwner()->getParentOp(); return value.getDefiningOp(); } // Check if the TPUCopyWithDynamicShapeOp is valid. // 1. The op should be wrapped inside a launch op. // 2. The wrapped launch op should be placed on CPU. LogicalResult CheckOpIsValid(Operation* op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/default_quant_params.cc
for (int non_bias : non_biases) { Operation *non_bias_define = op->getOperand(non_bias).getDefiningOp(); if (auto dequant = llvm::dyn_cast<TFL::DequantizeOp>(non_bias_define)) { auto non_bias_type = mlir::cast<TensorType>(dequant.getInput().getType()); auto non_bias_ele_type = mlir::cast<quant::QuantizedType>(non_bias_type.getElementType()); non_bias_types.push_back(non_bias_ele_type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/merge_fusion_with_dequantize.cc
PatternRewriter& rewriter) const override { if (call_op.getNumResults() != 1) return failure(); auto users = call_op->getUsers(); for (auto user : users) { if (!llvm::isa<mlir::stablehlo::UniformDequantizeOp>(user)) { return failure(); } } auto func_name = call_op.getCallee(); if (!func_name.starts_with("quantized_")) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_annotate_dynamic_shape_inputs.cc
shape[0] = ShapedType::kDynamic; auto extensions = mhlo::TypeExtensionsAttr::get(func->getContext(), bounds); auto resultType = RankedTensorType::get(shape, inputType.getElementType(), extensions); arg.setType(resultType); } llvm::SmallVector<Type, 8> arg_types; for (auto arg : func.getArguments()) arg_types.push_back(arg.getType()); func.setType(
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/utils/arithmetic_count_util.h
int64_t* count) { auto weight = op->getOperand(1); auto weight_type = mlir::dyn_cast_or_null<mlir::RankedTensorType>(weight.getType()); if (weight_type == nullptr || !weight_type.hasStaticShape()) return false; auto output = op->getResult(0); auto output_type = mlir::dyn_cast_or_null<mlir::RankedTensorType>(output.getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.1K bytes - Viewed (0)