- Sort Score
- Result 10 results
- Languages All
Results 71 - 73 of 73 for getElementDtype (0.34 sec)
-
tensorflow/compiler/mlir/lite/transforms/optimize_batch_matmul.cc
permuted_shape.begin() + input_rank - 2); return rewriter.create<TFL::TransposeOp>( bmm_op->getLoc(), RankedTensorType::get(permuted_shape, input_type.getElementType()), input, permutation_tensor_op.getResult()); }; Value input_lhs = bmm_op.getX(); Value input_rhs = bmm_op.getY(); Value output_lhs =
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/lite/stablehlo/odml_converter/transforms/outline_composites.cc
auto cst_op = llvm::dyn_cast_or_null<stablehlo::ConstantOp>(cst_input); if (!cst_op) return false; ElementsAttr value = cst_op.getValue(); if (!value.isSplat()) return false; if (!value.getElementType().isF32()) return false; return std::abs(value.getSplatValue<float>() - val) < kTolerance; } // Determines if the given op is semantically that of the gauss error function. bool MatchERF(Operation* op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool.cc
llvm::SmallVector<int64_t> shape = {pool.n, h, w, pool.c}; auto op_type = mlir::cast<RankedTensorType>(op->getResult(0).getType()); return RankedTensorType::get(shape, op_type.getElementType()); } StringAttr GetAvgPoolOpPadAttr(Builder& builder, CompositeOp op) { const TorchAvgPoolData pool = GetTorchAvgPoolData(op); if (pool.ph == 0 && pool.pw == 0) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 9.2K bytes - Viewed (0)