- Sort Score
- Result 10 results
- Languages All
Results 121 - 130 of 182 for created (0.11 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
rewriter.create<SubtractOp>(loc, b_max_diag_len, diag_len_d), b_zero); // x = max(d, 0) - offset // y = max(-d, 0) - offset Value x = rewriter.create<SubtractOp>( loc, rewriter.create<MaxOp>(loc, d, b_zero), offset); Value y = rewriter.create<SubtractOp>( loc, rewriter.create<MaxOp>(loc, neg_d, b_zero), offset); Value n_plus_x = rewriter.create<AddOp>(loc, iotaN, x);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor.cc
TF_RETURN_IF_ERROR(c_event->Create()); return std::move(c_event); } absl::StatusOr<std::unique_ptr<Stream>> CreateStream( std::optional<std::variant<StreamPriority, int>> priority = std::nullopt) override { auto stream = std::make_unique<CStream>(&device_, stream_executor_, this); TF_RETURN_IF_ERROR(stream->Create()); return std::move(stream); } private:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 14 07:39:19 UTC 2024 - 27.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_batch_matmul.cc
if (constant.getType().getRank() != 2) return failure(); // Create a tfl.transpose op that performs ZX transpose on `input`. auto create_z_x_transpose_op = [&](Value input) -> Value { RankedTensorType input_type = mlir::cast<RankedTensorType>(input.getType()); const int input_rank = input_type.getRank(); // Create a 1D I32 tensor for representing the dimension permutation.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util_test.cc
context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); mlir::OwningOpRef<mlir::ModuleOp> module_ref = mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); mlir::OpBuilder builder(module_ref->getBodyRegion()); llvm::SmallVector<mlir::Type, 8> result_types; auto cluster = builder.create<mlir::tf_device::ClusterOp>( mlir::UnknownLoc::get(&context), result_types); cluster->setAttr(kNumCoresPerReplicaAttr,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 46.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/nchw_convolution_to_nhwc.cc
Value input = op->getOperand(0); const TensorType new_input_tensor_type = GetTransposedTensorType( mlir::cast<TensorType>(input.getType()), kNchwToNhwcPermutation); auto input_transpose_op = rewriter.create<mlir::stablehlo::TransposeOp>( op.getLoc(), /*resultType0=*/new_input_tensor_type, /*operand=*/input, rewriter.getDenseI64ArrayAttr(kNchwToNhwcPermutation));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize.cc
/*quant_dim=*/-1, num_bits, narrow_range, is_signed); builder.setInsertionPoint(block, insertion_point); auto q_op = builder.create<quantfork::QuantizeCastOp>( loc, params.getValue(), arg); auto dq_op = builder.create<quantfork::DequantizeCastOp>( loc, input_type, q_op.getResult()); arg.replaceAllUsesWith(dq_op.getResult()); q_op.setOperand(arg); }
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/quantization/tensorflow/passes/prepare_lifting.cc
if (!dq_op) { auto mul_op = builder.create<TF::MulOp>(loc, value, multiplier); return mul_op.getResult(); } auto q_op = dq_op.getArg().getDefiningOp<quantfork::QuantizeCastOp>(); if (!q_op) return {}; Value float_value = q_op.getArg(); Value new_value = builder.create<TF::MulOp>(loc, float_value, multiplier);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/defer_activation_transpose.cc
} // Convenience function to create a `TransposeOp` with a given `permutation`. // The Location is set as `input`'s loc. TransposeOp CreateTransposeOp(Value input, const ArrayRef<int64_t> permutation, PatternRewriter& rewriter) { return rewriter.create<TransposeOp>( input.getLoc(), input, rewriter.getDenseI64ArrayAttr(permutation)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/decompose_hybrid_quantization.cc
if (QuantizedType::getQuantizedElementType(operand.getType())) { auto newTy = QuantizedType::castToExpressedType(operand.getType()); newOperands.push_back( rewriter.create<TFL::DequantizeOp>(loc, newTy, operand)); continue; } newOperands.push_back(operand); } SmallVector<Type> newResultTys; for (auto result : op->getResults()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc
if (globalTensor.getValue()) { initial_value = *globalTensor.getValue(); } else { initial_value = mlir::Attribute(); } opToName[globalTensor] = name; auto variableOp = globalBuilder.create<ml_program::GlobalOp>( globalTensor.getLoc(), name, globalTensor.getType(), globalTensor.getIsMutable(), initial_value, /*visibility=*/globalBuilder.getStringAttr("private"));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0)