- Sort Score
- Result 10 results
- Languages All
Results 101 - 110 of 141 for created (0.07 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_tf_xla_call_module_to_stablehlo_pass.cc
if (operand.getType() != expected_type) { operand = rewriter.create<TF::CastOp>( op.getLoc(), expected_type, operand, /*Truncate=*/rewriter.getBoolAttr(false)); } casted_operands.push_back(operand); } auto call = rewriter.create<func::CallOp>( op->getLoc(), main_fn.getSymName(), main_fn.getResultTypes(), casted_operands);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 25 09:43:18 UTC 2024 - 10.9K 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/tensorflow/utils/dump_mlir_util.cc
if (!status.ok()) { LOG(WARNING) << "cannot create directory '" << path << "': " << status.message(); return; } path += "/mlir_reproducer_"; if (!tensorflow::Env::Default()->CreateUniqueFileName(&path, ".mlir")) { LOG(WARNING) << "cannot create unique filename, won't enable MLIR crash " "reproducer."; return;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 05 03:03:46 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/fallback_to_flex_ops.cc
TF::Conv2DOp, TF::DepthwiseConv2dNativeOp, TF::Conv2DBackpropInputOp, TF::Conv3DOp, TF::Conv3DBackpropInputV2Op // clang-format on >(op); } // Creates the custom option of the Flex ops. inline void CreateFlexOpCustomOptions(const std::string &op_name, const std::string &node_def_str,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils.cc
} } // Create a dense constant op for slice's begin auto slice_i2c_begin = CreateI64DenseConst(builder, begin_shape, begin_values, location); // Create a dense constant op for slice's size auto slice_i2c_size = CreateI64DenseConst(builder, size_shape, size_values, location); return builder->create<TF::SliceOp>( location, RankedTensorType::get(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/prepare_tpu_computation_for_tf_export.cc
RankedTensorType::get({3}, rewriter.getType<TF::StringType>()); auto dynamic_key = rewriter.create<TF::_XlaCompileMlirPlaceholderProgramKeyOp>( func.getLoc(), /*program=*/result_type, llvm::ArrayRef<Value>{}); auto recv_at_host = rewriter.create<TF::_XlaRecvAtHostOp>( func.getLoc(), op.getOperandTypes(), /*dynamic_key=*/dynamic_key, op.getSendKeyAttr(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/push_transpose_through_ewise.cc
auto new_out_type = RankedTensorType::get( PermuteShape(current_out_type.getShape(), inverse_perm_arr), current_out_type.getElementType()); // Create new ewise op to appear before the tranpose. auto *new_ewise_op = rewriter.create(op->getLoc(), op->getName().getIdentifier(), {tpose_arg1.getOperand(0), tpose_arg2.getOperand(0)}, new_out_type, op->getAttrs());
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/stablehlo/transforms/legalize_stablehlo_to_vhlo.cc
std::optional<Value> MaterializeIllegalCast(OpBuilder &builder, Type type, ValueRange inputs, Location loc) { return builder.create<UnrealizedConversionCastOp>(loc, type, inputs) ->getResult(0); } class StablehloToOdmlTypeConverter : public vhlo::VhloTypeConverter { public: StablehloToOdmlTypeConverter() : vhlo::VhloTypeConverter() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 19:48:51 UTC 2024 - 12.3K bytes - Viewed (0)