- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 345 for SmallVector (0.15 sec)
-
tensorflow/compiler/mlir/tfrt/saved_model/saved_model.cc
// and input devices, variables used by this function. llvm::SmallVector<llvm::StringRef, 4> input_names; llvm::SmallVector< std::pair<tensorflow::DataType, tensorflow::PartialTensorShape>, 4> input_specs; llvm::SmallVector<llvm::StringRef, 4> input_devices; llvm::SmallVector<mlir::Operation*, 4> bound_inputs; for (unsigned i = 0, e = func.getNumArguments(); i != e; ++i) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.cc
#include "mlir/Support/LLVM.h" // from @llvm-project namespace mlir { namespace TF { SmallVector<int64_t, 4> ReversePermutation(ArrayRef<int64_t> permutation) { SmallVector<int64_t, 4> reverse(permutation.size()); for (size_t i = 0; i < permutation.size(); ++i) { reverse[permutation[i]] = i; } return reverse; } SmallVector<int64_t, 4> GetDataFormatPermutation(StringRef from, StringRef to) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/common/outline_operations.cc
for (Operation& op : nested_block->getOperations()) update_from_op(&op); }); } return SmallVector<Value>(results.getArrayRef()); } llvm::SmallVector<Value> AccumulateResultsDefinedWithin( const llvm::SetVector<Operation*>& partition_ops) { llvm::SmallVector<Value> values_for_results; for (Operation* op : partition_ops) { if (IsConstantOrNone(op)) { continue; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc
func::FuncOp createLstmCompositeFunc(mlir::Builder* builder, bool ln, bool cifg) { SmallVector<int64_t, 2> input_shape{1, 2}; SmallVector<int64_t, 2> weight_shape{3, 12}; SmallVector<int64_t, 1> bias_shape{2}; SmallVector<int64_t, 2> projection_shape{1, 2}; SmallVector<int64_t, 1> layer_norm_scale{4}; SmallVector<int64_t, 2> output_shape{1, 2}; auto input_type = RankedTensorType::get(input_shape, builder->getF32Type());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/constant_fold.cc
namespace mlir { namespace quant { namespace { // Folds the operation recursively and return the results. LogicalResult FoldOperation(OpBuilder& builder, Operation* op, SmallVector<Value>& results) { SmallVector<ElementsAttr> inputs; for (auto operand : op->getOperands()) { auto preceding_const_op = operand.getDefiningOp<TF::ConstOp>(); if (preceding_const_op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
assert(devices.find(tensorflow::GetDeviceAliasForLogicalCore(0)) ->getSecond() .size() == num_replicas); llvm::SmallVector<std::pair<ValueRange, Type>, 8> new_replicated_inputs; llvm::SmallVector<Value, 8> new_packed_inputs; llvm::SmallVector<llvm::SmallVector<Value, 8>, 8> replicated_inputs; replicated_inputs.reserve(replicate.GetNumReplicatedBlockArguments());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_colocate_composite_resource_ops.cc
builder->create<tf_device::ReturnOp>(loc, op->getResults()); // Move op inside cluster. op->moveBefore(launch.GetBody().getTerminator()); } llvm::SmallVector<Operation*, 4> GetResourceOpsUsingCompositeArgsInReplicate( tf_device::ReplicateOp replicate) { llvm::SmallVector<Operation*, 4> resource_users; const auto add_resource_op_to_list = [&resource_users](Operation* op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 16 17:41:12 UTC 2023 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc
// `tf.AssignVariableOp`s and `tf.Const`s. Collects and returns the // `tf.VarHandleOp`s that are initialized by these `tf.AssignVariableOp`s. SmallVector<TF::VarHandleOp> CollectVariableOps( func::FuncOp session_init_func) { SmallVector<TF::VarHandleOp> var_handle_ops{}; for (auto assign_variable_op : llvm::make_early_inc_range( session_init_func.getOps<TF::AssignVariableOp>())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc
AnonymousIteratorV3Op CreateIterator(OpBuilder builder, llvm::ArrayRef<Type> dataset_types, ReduceDatasetOp reduce_dataset) { llvm::SmallVector<Attribute, 2> shape_attrs; llvm::SmallVector<Attribute, 2> type_attrs; for (Type type : dataset_types) { shape_attrs.push_back( TF::ShapeAttr::get(builder.getContext(), mlir::cast<ShapedType>(type)));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/push_transpose_through_ewise.cc
}; // Compute the permutation that would take `arr` to the identity. llvm::SmallVector<int32_t> InvertPermutation(llvm::SmallVector<int32_t> arr) { llvm::SmallVector<int32_t> inverse_arr(arr.size()); for (int32_t i = 0; i < arr.size(); ++i) { inverse_arr[arr[i]] = i; } return inverse_arr; } llvm::SmallVector<int64_t> PermuteShape(llvm::ArrayRef<int64_t> shape,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.5K bytes - Viewed (0)