- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 677 for Auto (0.19 sec)
-
tensorflow/compiler/mlir/lite/transforms/legalize_hashtables.cc
} bool checkWhetherGraphHasValidStaticLookupTables(ModuleOp module) { auto hashtables = GetAllOps<TF::HashTableV2Op>(&module); // No needs to run the legalization patterns. if (hashtables.empty()) { return false; } for (auto hashtable : hashtables) { auto key_dtype = hashtable.getKeyDtype(); auto value_dtype = hashtable.getValueDtype();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/tfl_to_std.cc
func.walk([&](Operation* op) { b.setInsertionPoint(op); if (auto dq = llvm::dyn_cast<DequantizeOp>(op)) { auto dcast = b.create<quantfork::DequantizeCastOp>( dq.getLoc(), dq.getOutput().getType(), dq.getInput()); dq.getOutput().replaceAllUsesWith(dcast); dq.erase(); } else if (auto q = llvm::dyn_cast<QuantizeOp>(op)) { auto qcast = b.create<quantfork::QuantizeCastOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 02:50:01 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/unroll_batch_matmul.cc
PatternRewriter& rewriter) { int64_t shape_rank = shape.size(); auto shape_spec_type = RankedTensorType::get({shape_rank}, rewriter.getIntegerType(64)); Type resultType = RankedTensorType::get(shape, element_type); auto constant_attr = DenseElementsAttr::get(shape_spec_type, shape); auto shape_tensor = rewriter.create<TF::ConstOp>(loc, shape_spec_type, constant_attr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/translate_utils.cc
void PopulateTfVersions(mlir::ModuleOp module, const VersionDef& versions) { mlir::Builder b(module.getContext()); auto producer = b.getNamedAttr("producer", b.getI32IntegerAttr(versions.producer())); auto min_consumer = b.getNamedAttr( "min_consumer", b.getI32IntegerAttr(versions.min_consumer())); auto bad_consumers = b.getNamedAttr( "bad_consumers", b.getI32ArrayAttr(llvm::ArrayRef<int32_t>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor_test.cc
SP_Stream stream) -> void { auto custom_stream = static_cast<SP_Stream_st*>(stream); ASSERT_EQ(custom_stream->stream_id, 14); delete custom_stream; stream_deleted = true; }; StreamExecutor* executor = GetExecutor(0); ASSERT_FALSE(stream_created); TF_ASSERT_OK_AND_ASSIGN(auto stream, executor->CreateStream()); ASSERT_TRUE(stream_created);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 19:54:04 UTC 2024 - 26.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc
flattened_contracting_segids[i] = 0; } auto seg_prod_result_type = RankedTensorType::get(static_cast<int32_t>(1), builder.getI32Type()); auto out_segids_cst = builder.create<TFL::ConstOp>( builder.getI32TensorAttr(flattened_out_segids)); auto contracting_segids_cst = builder.create<TFL::ConstOp>( builder.getI32TensorAttr(flattened_contracting_segids)); auto num_segids_tensor =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/fold_broadcast_pass.cc
typename Convert> static Attribute BinaryFolder(Op *op) { auto lhs_op = op->getLhs().template getDefiningOp<mhlo::ConstantOp>(); auto rhs_op = op->getRhs().template getDefiningOp<mhlo::ConstantOp>(); if (!lhs_op || !lhs_op) return {}; auto lhs = dyn_cast_or_null<DenseElementsAttr>(lhs_op.getValue()); auto rhs = dyn_cast_or_null<DenseElementsAttr>(rhs_op.getValue()); if (!lhs || !rhs) return {};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/push_transpose_through_ewise.cc
} if (!tpose_arg->hasOneUse()) { return failure(); } auto tpose_arg_type = llvm::dyn_cast<RankedTensorType>(tpose_arg->getResultTypes()[0]); auto cst_arg_type = llvm::dyn_cast<RankedTensorType>(cst_arg->getResultTypes()[0]); auto tpose_arg_rank = tpose_arg_type.getRank(); auto cst_arg_rank = cst_arg_type.getRank();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_composite_functions_tf.cc
size_t start_map = fbb.StartMap(); for (auto attr : attrs) { if (auto float_attr = mlir::dyn_cast_or_null<FloatAttr>(attr.second)) { fbb.Float(attr.first.data(), float_attr.getValue().convertToFloat()); } else if (auto int_attr = mlir::dyn_cast_or_null<IntegerAttr>(attr.second)) { fbb.Int(attr.first.data(), int_attr.getInt()); } else if (auto bool_attr = mlir::dyn_cast_or_null<BoolAttr>(attr.second)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc
SmallVector<TransposeOp, 2>* transpose_ops) { for (auto it = transpose_ops->begin(); it != transpose_ops->end(); ++it) { auto tranpose_op = *it; for (auto tranpose_operand : tranpose_op.getOperands()) { auto ranked_tranpose_type = mlir::dyn_cast_or_null<RankedTensorType>(tranpose_operand.getType()); if (!ranked_tranpose_type) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.3K bytes - Viewed (0)