- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for resultType (0.18 sec)
-
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
llvm::function_ref<APFloat(APFloat)> calculate) { assert(IsF32ShapedType(result_type) || IsBF16ShapedType(result_type)); auto result_shape_type = result_type.cast<ShapedType>(); if (!result_shape_type.hasStaticShape()) return {}; if (auto dense_elements = operand.dyn_cast_or_null<DenseElementsAttr>()) { SmallVector<APFloat, 16> new_values;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
def TFL_BroadcastableBinaryBuilder : OpBuilder<(ins "Value":$lhs, "Value":$rhs), [{ auto resultType = OpTrait::util::getBroadcastedType(lhs.getType(), rhs.getType()); if (!resultType) mlir::emitError($_state.location, "non-broadcastable operands"); $_state.addOperands({lhs, rhs}); $_state.types.push_back(resultType); }]>; def TFL_FusedBroadcastableBinaryBuilder :
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
Type operand_type = std::get<0>(entry).getType(); Value result = std::get<1>(entry); TensorType result_type = mlir::cast<TensorType>(result.getType()); Type inferred_type = TypeMeet(result_type, operand_type); if (result_type == inferred_type) continue; if (!UpdateTypeAndInsertIncompatibleUseCasts(inferred_type, result)) continue; changed = true; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
auto matmul = rewriter.create<TF::BatchMatMulV3Op>( loc, RankedTensorType::get(matmul_shape, result_type.getElementType()), lhs_flattend, rhs_flattend); if (result_type.hasStaticShape()) { auto reshaped = rewriter.create<mhlo::ReshapeOp>(loc, result_type, matmul.getResult()); return reshaped.getResult(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
Value y, BoolAttr incompatible_shape_error) { auto result_type = DeduceEqualCmpOpType(&builder, result.location, x, y, incompatible_shape_error); return build(builder, result, result_type, x, y, incompatible_shape_error); } //===----------------------------------------------------------------------===// // OneHotOp
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
for (auto it : llvm::zip(op.getResultTypes(), updated_results)) { auto& updated_result = std::get<1>(it); Type result_type = std::get<0>(it); if (result_type != updated_result.getType()) { updated_result = rewriter.create<TF::CastOp>(op.getLoc(), result_type, updated_result, /*Truncate=*/rewriter.getBoolAttr(false)); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
auto indices_value = gather_nd_first.getIndices(); auto result_type = dyn_cast<RankedTensorType>(result_value.getType()); auto params_type = dyn_cast<RankedTensorType>(params_value.getType()); auto indices_type = dyn_cast<RankedTensorType>(indices_value.getType()); // Handle static shape cases only. if (!result_type || !params_type || !indices_type) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
return success(); } void rewrite(stablehlo::UniformDequantizeOp op, PatternRewriter& rewriter) const override { rewriter.replaceOpWithNewOp<TFL::DequantizeOp>( op, /*resultTypes=*/op->getResultTypes(), /*input=*/op.getOperand()); } }; // Rewrites `stablehlo.dot_general` to `tfl.fully_connected` or // `tfl.batch_matmul` when it accepts uniform quantized tensors. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
os << operand_type; } } os << ") -> ("; if (!inst->getResultTypes().empty()) { bool first = true; for (Type result_type : inst->getResultTypes()) { os << (!first ? ", " : ""); first = false; os << result_type; } } os << ")"; // Print out attributes except for large elementsattributes (which should
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)