- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for sizeType (0.24 sec)
-
src/cmd/compile/internal/ssa/memcombine.go
if needSwap && (size != 1 || !root.Block.Func.Config.haveByteSwap(n)) { return false } // This is the commit point. // First, issue load at lowest address. v = loadBlock.NewValue2(pos, OpLoad, sizeType(n*size), r[0].load.Args[0], mem) // Byte swap if needed, if needSwap { v = byteSwap(loadBlock, pos, v) } // Extend if needed. if n*size < root.Type.Size() {
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Thu Mar 21 19:45:41 UTC 2024 - 18.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc
} Value ReshapeScalarToSizeType(OpBuilder builder, Value scalar, Location loc) { auto size_type = GetSizeType(builder); return builder.create<TF::ReshapeOp>( loc, ArrayRef<Type>{size_type}, ArrayRef<Value>{scalar, GetR1Const(size_type.getShape(), builder, loc)}); } LogicalResult CreateInitBufferValue(ArrayRef<int64_t> element_shape,
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/lite/experimental/tac/transforms/device_transform_patterns.cc
if (!begin_type || !begin_type.hasStaticShape()) return failure(); auto size = slice_op.getSize(); auto size_type = mlir::dyn_cast_or_null<RankedTensorType>(size.getType()); if (!size_type || !size_type.hasStaticShape()) return failure(); auto output_type = mlir::dyn_cast_or_null<RankedTensorType>(slice_op.getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc
void runOnOperation() final; }; // Returns the type of the local variable for the stack size. Type GetSizeVarType(OpBuilder builder) { auto size_type = cutil::GetSizeType(builder); return RankedTensorType::get( {}, TF::ResourceType::get(ArrayRef<TensorType>{size_type}, builder.getContext())); } // Returns the aliasing argument number of a fucntion return value if it simply
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
struct SizeInfo { Value size; bool fixed; }; // Modifies a function's signature to rewrite tensor list arguments to buffers // and sizes. void ModifyFunctionSignature( func::FuncOp func, Type size_type, llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, llvm::function_ref<std::optional<Type>(int64_t)> arg_to_buffer_type, llvm::function_ref<bool(int64_t)> arg_buffer_size_is_fixed) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
slice_op.setOperand(1, new_begin); return success(); } } // Handle size. if (size_op && size_type && size_type.getElementType().isInteger(64)) { auto new_size = NarrowDownInt64InputValuesForOp( size_op, size_type, slice_op.getLoc(), &rewriter); if (new_size != nullptr) { slice_op.setOperand(2, new_size); return success(); }
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/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
lower_limit_cst); SmallVector<int64_t> size_len{rank}; TensorType size_type = operand_type.cloneWith(size_len, i64_type); auto size_attr = DenseIntElementsAttr::get(size_type, slice_sizes); auto size = rewriter.create<arith::ConstantOp>(op.getLoc(), size_attr); rewriter.replaceOpWithNewOp<TFL::SliceOp>(op, output, input, begin, size);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
auto begin_type = mlir::dyn_cast<RankedTensorType>(begin_indices.getType()); auto size_type = mlir::dyn_cast<RankedTensorType>(sizes.getType()); if (!input_ty || !begin_type || !size_type || !begin_type.hasStaticShape() || !size_type.hasStaticShape() || begin_type.getRank() != 1 || size_type.getRank() != 1) { return failure(); }
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/transforms/shape_inference.cc
func::FuncOp ret = queue_.front(); queue_.pop(); queue_set_.erase(ret); return ret; } // Returns the current size of the queue. std::queue<func::FuncOp>::size_type QueueSize() const { return queue_.size(); } Dialect* const tf_dialect_; private: // Returns whether the result of an operation could be updated to a new
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)