- Sort Score
- Result 10 results
- Languages All
Results 111 - 120 of 129 for created (0.07 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize.cc
rewriter.setInsertionPointAfter(preceding_sc_op); auto fcast_op = rewriter.create<TF::CastOp>( preceding_sc_op->getLoc(), dq_arg_type.clone(rewriter.getF32Type()), preceding_sc_op.getResult()); // Create a new AvgPool op with float type. TF::AvgPoolOp float_avg_pool_op = rewriter.create<TF::AvgPoolOp>( avg_pool_op->getLoc(), avg_pool_op.getType().clone(rewriter.getF32Type()),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 23.6K bytes - Viewed (0) -
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/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/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/transforms/post_quantize.cc
TFL::populateWithGenerated(patterns); patterns.add<RemoveVolatileOps<kPreserveNone>>(ctx); (void)applyPatternsAndFoldGreedily(func, std::move(patterns)); } } // namespace // Creates an instance of the TensorFlow Lite dialect PostQuantize pass. std::unique_ptr<OperationPass<func::FuncOp>> CreatePostQuantizePass( bool emit_quant_adaptor_ops, const quant::CustomOpMap& custom_op_map) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/add_dump_tensor_op.cc
const StringAttr &new_ref_func_name) { // Create PartitionedCallOp to the copied composite function. This // PartitionedCallOp does not have kQuantTraitAttrName, and therefore won't // get quantized. auto new_call_op = rewriter.create<TF::PartitionedCallOp>( call_op.getLoc(), call_op.getResultTypes(), call_op.getOperands(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 22:55:22 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
if (!tensor_attr_or.ok()) return nullptr; return tensor_attr_or.value(); } // Creates a constant op that holds 'tensor_elements'. TF::ConstOp GetConstOpFromElementsAttr(ElementsAttr tensor_elements, OpBuilder builder, Location loc) { return builder.create<TF::ConstOp>(loc, tensor_elements); } // Returns ElementsAttr which has the value held by 'resource_tensor'.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize.cc
mlir::dyn_cast<ShapedType>(v.getType()).getShape()[rhs_batch_dim]; } auto lhs_new_concat = rewriter.create<mhlo::ConcatenateOp>( concat->getLoc(), concat.getType().clone(lhs_new_concat_shape), all_dot_lhs, rewriter.getI64IntegerAttr(lhs_batch_dim)); auto rhs_new_concat = rewriter.create<mhlo::ConcatenateOp>( concat->getLoc(), concat.getType().clone(rhs_new_concat_shape),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 26.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_dynamic_layout_pass.cc
Value compilation_key, tf_device::LaunchOp compile_launch, OpBuilder* builder) { return builder->create<TF::TPUGetLayoutOp>( compile_launch.getLoc(), llvm::ArrayRef<Type>{RankedTensorType::get({ShapedType::kDynamic}, builder->getIntegerType(64))},
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.7K bytes - Viewed (0)