- Sort Score
- Result 10 results
- Languages All
Results 151 - 160 of 1,825 for Auto (0.07 sec)
-
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/tf2xla/internal/passes/tpu_cluster_formation.cc
llvm::SmallVector<mlir::TF::IdentityOp, 8> erase_list; for (auto [cluster_result_id, cluster_result] : llvm::enumerate(cluster.getResults())) { auto [replicated_output, _] = GetSingleUserOfType<mlir::TF::TPUReplicatedOutputOp>(cluster_result); if (replicated_output) { for (OpResult per_replica_result : replicated_output->getResults()) { auto [partitioned_output, id_ops] =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K 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/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) -
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/tensorflow/ir/tf_ops_a_m.cc
auto type = mlir::cast<ShapedType>(getResult().getType()); auto etype = type.getElementType(); // We can not fold if the result is not static. if (!type.hasStaticShape()) return {}; if (auto float_type = mlir::dyn_cast<FloatType>(etype)) { auto out_type = tensorflow::GetTypeFromTFTensorShape(out_shape, float_type); return DenseElementsAttr::get(out_type,
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/jit/encapsulate_util.cc
OutsideCompilationClusterDependencies( const Graph* g, const string& outside_compilation_attr_name) { auto cluster_deps = std::make_unique< absl::flat_hash_map<string, absl::flat_hash_set<string>>>(); for (const Edge* e : g->edges()) { auto src_outside_compilation = GetStringAttr(*e->src(), outside_compilation_attr_name); auto dst_outside_compilation = GetStringAttr(*e->dst(), outside_compilation_attr_name);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.5K 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) -
tensorflow/compiler/mlir/quantization/stablehlo/ops/stablehlo_op_quant_spec_test.cc
ASSERT_TRUE(module_op); auto test_func = module_op->lookupSymbol<func::FuncOp>("same_scale_after_composite"); ASSERT_THAT(test_func, NotNull()); auto quantize_op = FindOperationOfType<quantfork::QuantizeCastOp>(test_func); EXPECT_FALSE(IsOpQuantizableStableHlo(quantize_op)); auto dequantize_op = FindOperationOfType<quantfork::DequantizeCastOp>(test_func);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 04 07:19:09 UTC 2024 - 14.8K 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)