- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 171 for getElementDtype (0.49 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize.cc
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/lite/stablehlo/transforms/legalize_hlo.cc
if (!keys_ty || !keys_ty.hasStaticShape() || !keys_ty.getElementType().isIntOrFloat()) return rewriter.notifyMatchFailure( op, "only match for the case where the first operand has a static " "int/float shapeType"); if (!indices_ty || !indices_ty.hasStaticShape() || !indices_ty.getElementType().isInteger(32)) return rewriter.notifyMatchFailure( op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
platforms/core-configuration/model-core/src/main/java/org/gradle/model/internal/manage/schema/extract/SpecializedMapNodeInitializer.java
modelNode.addProjection(new SpecializedModelMapProjection<T, E>(schema.getType(), schema.getElementType(), implementationType, strategyAccessor)); modelNode.addProjection(ModelMapModelProjection.managed(schema.getType(), schema.getElementType(), strategyAccessor)); } } ))
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Thu Sep 28 09:51:04 UTC 2023 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc
RankedTensorType::get(lhs_flattened_shape, lhs_type.getElementType()), lhs_transposed.getResult()); } else { auto lhs_flattend_shape_op = BuildDotOperandFlattenedShapeOp( lhs, lhs_dot_dimensions_info, builder, /*is_lhs=*/true); lhs_flattend = rewriter.create<mhlo::DynamicReshapeOp>( loc, RankedTensorType::get(lhs_flattened_shape, lhs_type.getElementType()), lhs_transposed, lhs_flattend_shape_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/validators.h
if (!tensorType) return false; return tensorType.getElementType().isF32(); } // Returns true iff the given value is a bf16 tensor. inline bool TFTypeIsBFloat16Tensor(Value value) { auto tensorType = mlir::dyn_cast<TensorType>(value.getType()); if (!tensorType) return false; return tensorType.getElementType().isBF16(); } // Returns true iff the given value is a f16 tensor.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
const auto output_type = getType(0).cast<ShapedType>(); // Folding only implemented for float tensors. if (!input_type.getElementType().isF32() || !weights_type.getElementType().isF32() || !output_type.getElementType().isF32() || (has_bias && !bias_type.getElementType().isF32())) { return failure(); } // Folding only implemented for static shapes
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/tensorflow/transforms/decompose_resource_ops.cc
.getElementType() .cast<ResourceType>(); if (resource_type.getSubtypes().size() == 1) return resource_type.getSubtypes().front(); return UnrankedTensorType::get(element_type); } static bool HasResourceSubtype(Value resource) { return resource.getType() .cast<TensorType>() .getElementType() .cast<ResourceType>()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/rewrite_util.h
IntegerAttr attr = IntegerAttr::get(int_ty, raw_value); return DenseElementsAttr::get(scalar_ty, attr); } else if (auto complex_ty = mlir::dyn_cast<ComplexType>(ty)) { Type complex_element_ty = complex_ty.getElementType(); if (complex_element_ty.isF32()) { return DenseElementsAttr::get( scalar_ty, static_cast<std::complex<float>>(raw_value)); } else if (complex_element_ty.isF64()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc
result_shape, buffer_type.getElementType()); return builder.create<TF::SliceOp>( loc, ArrayRef<Type>{slice_type}, ArrayRef<Value>{buffer, GetR1Const(slice_starts, builder, loc), GetR1Const(result_shape, builder, loc)}); } auto result_type = tensorflow::GetTypeFromTFTensorShape( result_shape, buffer_type.getElementType()); return builder.create<TF::GatherV2Op>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
if (mlir::isa<QuantizedType>(type.getElementType())) { has_quantized_types = true; } } } for (Value output : call_op.getOutput()) { if (auto type = mlir::dyn_cast<TensorType>(output.getType())) { if (mlir::isa<QuantizedType>(type.getElementType())) { has_quantized_types = true; } } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0)