- Sort Score
- Result 10 results
- Languages All
Results 81 - 90 of 169 for getElementDtype (0.32 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc
redux_axes.end()) { sum_shape.push_back(shape[i]); } } return rewriter->create<TF::SumOp>( loc, RankedTensorType::get(sum_shape, value_type.getElementType()), value, redux_op); } TF::TransposeOp createTransposeOp(Value value, Location loc, llvm::ArrayRef<int32_t> permutation,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/common/utils.h
// Returns true if it is a shaped type of f32 elements. inline bool IsF32ShapedType(Type t) { if (auto shaped_type = mlir::dyn_cast_or_null<ShapedType>(t)) { return shaped_type.getElementType().isF32(); } return false; } // Return true when the given element_type is QI8. inline bool IsQI8Type(Type t) { auto quantized_type = quant::QuantizedType::getQuantizedElementType(t);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_partitioning.cc
.getElementType() .cast<TF::ResourceType>() .getSubtypes())) return false; return true; } Type GetResourceSubtype(Type type) { return type.cast<TensorType>() .getElementType() .cast<TF::ResourceType>() .getSubtypes() .front(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc
elem_shape.push_back(length); for (int64_t dim : buffer_type.getShape().drop_front()) { elem_shape.push_back(dim); } *elem_type = RankedTensorType::get(elem_shape, buffer_type.getElementType()); return success(); } // Tries to infer the tensor array element shape. std::optional<llvm::SmallVector<int64_t, 8>> GetTensorArrayElementShape( TF::TensorArrayV3Op ta, ModuleOp module) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 40.2K bytes - Viewed (0) -
platforms/core-configuration/model-core/src/main/java/org/gradle/api/internal/provider/CollectionPropertyInternal.java
import java.util.Collection; public interface CollectionPropertyInternal<T, C extends Collection<T>> extends PropertyInternal<C>, HasMultipleValues<T>, CollectionProviderInternal<T, C> { @Override Class<T> getElementType(); /** * Adds an element to the property value. * * <p> * When invoked on a property with no value, this method first sets the value
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Thu May 23 18:32:13 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.cc
const int axis) { const auto source_type = dyn_cast_or_null<ShapedType>(source.getValue()); if (!source_type) return {}; const auto src_ele_type = source_type.getElementType(); auto qtype = dyn_cast<quant::QuantizedType>(src_ele_type); // Reset the quantization dimensions if it is per-axis. if (const auto per_axis =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 02:10:16 UTC 2024 - 43.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/hoist_replicate_invariant_resource_writes.cc
// similar. Move to common utils. bool IsResourceType(Type type) { return type.isa<TF::ResourceType>() || (type.isa<TensorType>() && type.cast<TensorType>().getElementType().isa<TF::ResourceType>()); } SmallVector<Value> GetAccessedResources(Operation& op) { SmallVector<Value, 4> accessed_resources; for (auto operand : op.getOperands()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_tensor_helper.h
// given `rank`. inline bool IsOfRankedFloatTensorType(RankedTensorType type, int rank) { return type && type.getRank() == rank && mlir::isa<FloatType>(type.getElementType()); } // Returns true if the given `value` has the specified rank or has unranked // type. inline bool IsOfRankOrUnranked(Value value, int64_t rank) { RankedTensorType type = GetRankedTensorTypeForOperand(value);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.8K bytes - Viewed (0) -
platforms/core-runtime/internal-instrumentation-processor/src/main/java/org/gradle/internal/instrumentation/processor/codegen/groovy/SignatureTree.java
// In the end, match the vararg parameter, if it is there: varargParameter.map(parameterInfo -> Stream.of(new ParameterMatchEntry(parameterInfo.getParameterType().getElementType(), VARARG))).orElseGet(Stream::empty) ).flatMap(Function.identity()).collect(Collectors.toList()); }
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Mon Oct 02 15:44:14 UTC 2023 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/folders.cc
res_attr); return success(); } static LogicalResult FoldDivOp(stablehlo::DivOp op, PatternRewriter& rewriter) { auto etype = op.getType().getElementType(); if (etype.isa<FloatType>()) { return FoldDivOpInternal<APFloat>(op, rewriter); } if (etype.isa<IntegerType>()) { return FoldDivOpInternal<APInt>(op, rewriter); } return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 06:11:55 UTC 2024 - 4.5K bytes - Viewed (0)