- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 140 for getElementDtype (0.41 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
size.getSExtValue(), start.getType().cast<TensorType>().getElementType()), start, limit, delta); } return RangeOp::build( builder, result, tensorflow::GetTypeFromTFTensorShape( {-1}, start.getType().cast<TensorType>().getElementType()), start, limit, delta); } OpFoldResult RangeOp::fold(FoldAdaptor adaptor) {
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/quantization/stablehlo/passes/quantize_weight.cc
// Non-float tensors do not need quantization. QuantizationUnits quantizable_ops; const ShapedType type = mlir::dyn_cast<ShapedType>(op.getType()); if (!type || !type.getElementType().isF32()) return quantizable_ops; const Value value = op.getResult(); for (OpOperand& use : value.getUses()) { Operation* user = use.getOwner();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.9K bytes - Viewed (0) -
platforms/core-runtime/internal-instrumentation-processor/src/main/java/org/gradle/internal/instrumentation/processor/codegen/TypeUtils.java
} if (type.equals(Type.DOUBLE_TYPE)) { return ClassName.DOUBLE; } if (type.getSort() == Type.ARRAY) { return ArrayTypeName.of(typeName(type.getElementType())); } return className(type); } public static ClassName className(Type type) { return ClassName.bestGuess(type.getClassName().replace("$", ".")); }
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Fri Apr 05 15:45:10 UTC 2024 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
mlir::dyn_cast<TensorFlowTypeWithSubtype>(operand_type.getElementType()); if (!operand_handle_type) return result_type.getElementType(); auto result_handle_type = mlir::cast<TensorFlowTypeWithSubtype>(result_type.getElementType()); if (operand_handle_type.GetSubtypes().empty() || !result_handle_type.GetSubtypes().empty()) return result_type.getElementType(); return operand_handle_type; }
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/composite_utils.cc
output_shape[3] = composite_result_shape[1]; auto input_type = mlir::cast<ShapedType>(old_op->getOperand(0).getType()); return RankedTensorType::get(output_shape, input_type.getElementType()); } } // namespace odml
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 18:33:05 UTC 2024 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/scatter.h
rewriter.getI32Type()); new_indices = rewriter.create<TF::ReshapeOp>( loc, RankedTensorType::get({num_updates, index_depth}, indices_type.getElementType()), indices, indices_shape); auto updates_shape = BuildIntArrayConstOp(builder, rewriter, llvm::SmallVector<int64_t>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/default_quant_params.cc
// doesn't require quantization. auto tensor_type = mlir::dyn_cast<TensorType>(value.getType()); if (!tensor_type) { // There are none type values. return; } if (!tensor_type.getElementType().isF32()) return; // If the result is consumed by a quantize op, it has been quantized. if (value.hasOneUse() && llvm::isa<TFL::QuantizeOp>(*value.getUsers().begin())) return;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op.cc
mlir::dyn_cast<DenseElementsAttr>(Quantize(attr, shaped_quantized_type)); if (!tensor_proto_attr) { return nullptr; } Type storage_type = mlir::cast<QuantizedType>(shaped_quantized_type.getElementType()) .getStorageType(); ShapedType new_type = shaped_quantized_type.clone(storage_type); rewriter.setInsertionPointAfter(op); auto const_op =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.h
if (mlir::isa<NoneType>(operand_type)) { inputs.push_back(operand); continue; } auto ele_type = mlir::cast<TensorType>(operand.getType()).getElementType(); if (auto dq_op = dyn_cast_or_null<DequantizeOpT>(operand.getDefiningOp())) { inputs.push_back(dq_op.getOperand()); } else if (!ele_type.isF32()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/unfold_splat_constant_pass.cc
if (!splat_elements_attr) { return; } if (splat_elements_attr.getNumElements() == 1) { return; } auto element_type = splat_elements_attr.getType().getElementType(); if (mlir::isa<ComplexType>(element_type) || mlir::isa<quant::QuantizedType>(element_type)) { return; } op_builder->setInsertionPoint(const_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.8K bytes - Viewed (0)