- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for SmallVectorImpl (0.38 sec)
-
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
void UpdateTensorListTypes( const llvm::SmallSet<int, 4> &tensor_list_index, const llvm::SmallSet<int, 4> &resized_tensor_list_index, ArrayRef<Type> types, R &&range, ValueRange operands, llvm::SmallVectorImpl<Type> *updated_types) { int i = 0; for (const auto it : llvm::zip(types, range, operands)) { if (tensor_list_index.count(i)) { // Only change the tensorlist's type to unranked tensor if it has been
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
return GetTensorFeatureDimIndex(input_ty.getRank(), format); } // Gets all integer values from the given attribute and push them to `values`. void GetI64ArrayAttrValues(Attribute attr, SmallVectorImpl<int64_t> *values) { auto array_attr = mlir::cast<ArrayAttr>(attr); values->reserve(array_attr.getValue().size()); for (Attribute val : array_attr.getValue())
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
// values, respectively. bool GetSlicedShapeAndBoundRanges( ::llvm::SmallVectorImpl<int64_t> *input_shape, ::llvm::SmallVectorImpl<int64_t> *slice_begin, ::llvm::SmallVectorImpl<int64_t> *slice_end, ::llvm::SmallVectorImpl<int64_t> *slice_stride); }]; } def TF_StringJoinOp : TF_Op<"StringJoin", [Pure]> { let summary = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
absl::flat_hash_map<ValuePort, Attribute, ValuePortHasher>; using ComputedQueryFn = function_ref<bool(ValuePort)>; using ValueQueryFn = function_ref<Attribute(const ValuePort&)>; using ValuePortInputs = SmallVectorImpl<ValuePort>; // Note: Following implements the rank 1 pack op case so could be // generalized. // // Maps the specified component in the `port` of the given op's result to one of // the element in the input.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)