- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 79 for getElementDtype (0.19 sec)
-
subprojects/core/src/test/groovy/org/gradle/api/internal/collections/ListElementSourceTest.groovy
final List<T> value TypedProviderOfList(Class<T> type, List<T> value) { this.type = type this.value = value } @Override Class<T> getElementType() { return type } @Override protected Value<List<T>> calculateOwnValue(ValueConsumer consumer) { return Value.of(value) }
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Mon Apr 13 05:06:48 UTC 2020 - 12.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
elemType.isInteger(64)) return true; // Complex<F<32>> is allowed. if (mlir::isa<ComplexType>(elemType) && mlir::cast<ComplexType>(elemType).getElementType().isF32()) return true; // QUINT8 and UI8 are allowed. if (mlir::isa<TF::Quint8Type>(elemType) || (elemType.isInteger(8) && mlir::cast<IntegerType>(elemType).isUnsigned())) return true;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/utils.h
for (int64_t i = 0; i < permutation_array.size(); ++i) { transposed_shape[i] = input_type.getDimSize(permutation_array[i]); } auto transposed_type = RankedTensorType::get(transposed_shape, input_type.getElementType()); return transposed_type; } // Returns shape of a ranked tensor. // Precondition: output_val's is ranked tensor. // Returns a truncated shape when `truncate` is set to true.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.cc
Type attribute_type = builder.getI64Type(); return builder.create<TF::StridedSliceOp>( loc, RankedTensorType::get( {}, mlir::cast<ShapedType>(shape_value.getType()).getElementType()), /*input=*/shape_value, /*begin=*/Create1DConstValue<int32_t>(builder, loc, {dim}), /*end=*/Create1DConstValue<int32_t>(builder, loc, {dim + 1}),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
subprojects/core/src/main/java/org/gradle/api/internal/collections/AbstractIterationOrderRetainingElementSource.java
} return element; } Element<T> cachingElement(CollectionProviderInternal<T, ? extends Iterable<T>> provider) { final Element<T> element = new Element<>(provider.getElementType(), new ElementsFromCollectionProvider<>(provider), this::doAddRealized); if (provider instanceof ChangingValue) {
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Mon Mar 04 20:04:06 UTC 2024 - 13.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tf_xla_op_to_tf_op.cc
// dimension with size 1. output_shape[i] = 1; } else { output_shape[i] = *shape_itr; shape_itr++; } } return RankedTensorType::get(output_shape, tensor_type.getElementType()); } // Determines the output type of the `SliceOp` when it is being inserted in // place of a `XlaGatherOp`. When the dimensions of `xla_gather_op_output_type`
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_traits.h
for (int i = 0, e = rank; i != e; i++) { int64_t dim0 = a.getDimSize(i); int64_t dim1 = b.getDimSize(i); dims[i] = (dim0 == ShapedType::kDynamic) ? dim1 : dim0; } return RankedTensorType::get(dims, a.getElementType()); } } // namespace detail // Verifies that op has the same operand and result element types (or type // itself, if scalar) after resolving reference types (i.e., after converting
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types.cc
Type ToLegalType(Type type) { if (IsTFQintType(type)) return GetIntTypeFromTFQint(type); if (auto shaped = mlir::dyn_cast<ShapedType>(type)) { Type elem = shaped.getElementType(); if (IsTFQintType(elem)) return shaped.clone(ToLegalType(elem)); } return type; } bool IsQintToIntCast(Operation *op) { auto cast_op = llvm::dyn_cast<TF::CastOp>(op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions.cc
} // Only the composite functions with f32 inputs are quantizable. if (call_op.getResults().size() == 1 && !mlir::cast<ShapedType>(call_op->getResult(0).getType()) .getElementType() .isF32()) { check_status.Update(absl::InternalError( "Composite functions for quantization should be f32 type.")); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 16.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops.td
def CreateTFReadVariableOp : NativeCodeCall< "$_builder.create<TF::ReadVariableOp>(" " $0.getLoc()," " GetResourceSubtypeOrDefault(" " $2, $1.getType().cast<TensorType>().getElementType())," " $2)" >; def CheckHasResourceSubtype : Constraint<CPred<"HasResourceSubtype($0)">>; def CreateConstBoolAttrFalse : NativeCodeCall<"$_builder.getBoolAttr(false)">;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 20.7K bytes - Viewed (0)