- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 19 for input_list (0.29 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
mlir::tf_device::ClusterFuncOp cluster_func, mlir::OpBuilder* builder, llvm::SmallVectorImpl<llvm::SmallVector<mlir::Value, 4>>* input_list) { // Initialize the input list for each logical devices. input_list->reserve(num_cores_per_replica); for (int i = 0; i < num_cores_per_replica; ++i) input_list->emplace_back(llvm::SmallVector<mlir::Value, 4>()); llvm::SmallVector<mlir::Value, 4> cluster_func_inputs(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
// i-th logical core. llvm::SmallVector<llvm::SmallVector<mlir::Value, 4>, 4> input_list; builder->setInsertionPoint(*new_parallel_execute); auto result = tensorflow::ExtractInputsForLogicalDevices( num_cores_per_replica, cluster_func, builder, &input_list); if (failed(result)) return failure(); const bool replicated = tpu_devices.size() != 1;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0) -
tensorflow/cc/ops/while_loop.cc
next_output_index, dtype); std::vector<NodeBuilder::NodeOut> input_list({enter_input, next_input}); const string unique_name = scope.GetUniqueNameForOp("Merge"); NodeBuilder builder = NodeBuilder(unique_name, "Merge").Input(input_list); scope.UpdateBuilder(&builder); Node* merge_node; TF_RETURN_IF_ERROR(builder.Finalize(scope.graph(), &merge_node));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 01:01:21 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.h
mlir::LogicalResult ExtractInputsForLogicalDevices( int num_cores_per_replica, mlir::tf_device::ClusterFuncOp cluster_func, mlir::OpBuilder* builder, llvm::SmallVectorImpl<llvm::SmallVector<mlir::Value, 4>>* input_list); // Extracts a list of OpSharding that represent output sharding configuration of // `tf_device.cluster`. mlir::LogicalResult ParseAndValidateOutputSharding(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 22:18:34 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
} // Creates a slice of the tensorlist `input_list`, starting from // [start_index, 0, ...0], with size [size, -1, ...-1]. // // Requires that `start_index` and `size` are scalar tensors and // `item_position_shape` is a 1-D tensor with only one element equal to the rank // of an item in the tensorlist. TF::SliceOp CreateSliceOpForTensorList(Location loc, Value input_list,
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/c/c_api.cc
int num_inputs) { std::vector<NodeBuilder::NodeOut> input_list; input_list.reserve(num_inputs); for (int i = 0; i < num_inputs; ++i) { input_list.emplace_back(&inputs[i].oper->node, inputs[i].index); } desc->node_builder.Input(input_list); } void TF_AddControlInput(TF_OperationDescription* desc, TF_Operation* input) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/cc/framework/ops.h
class InputList { public: /// Implicitly convert a list of outputs to a list of inputs. This is useful /// to write code such as ops::Concat(ops::Split(x, 4)). InputList(const OutputList& out) { // NOLINT(runtime/explicit) for (auto const& x : out) { inputs_.push_back(x); } } InputList(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 05:57:22 UTC 2024 - 10.5K bytes - Viewed (0) -
android/guava/src/com/google/common/collect/Collections2.java
final ImmutableList<E> inputList; final Comparator<? super E> comparator; final int size; OrderedPermutationCollection(Iterable<E> input, Comparator<? super E> comparator) { this.inputList = ImmutableList.sortedCopyOf(comparator, input); this.comparator = comparator; this.size = calculateSize(inputList, comparator); } /**
Registered: Wed Jun 12 16:38:11 UTC 2024 - Last Modified: Mon Apr 01 16:15:01 UTC 2024 - 22.8K bytes - Viewed (0) -
guava/src/com/google/common/collect/Collections2.java
final ImmutableList<E> inputList; final Comparator<? super E> comparator; final int size; OrderedPermutationCollection(Iterable<E> input, Comparator<? super E> comparator) { this.inputList = ImmutableList.sortedCopyOf(comparator, input); this.comparator = comparator; this.size = calculateSize(inputList, comparator); } /**
Registered: Wed Jun 12 16:38:11 UTC 2024 - Last Modified: Mon Apr 01 16:15:01 UTC 2024 - 23.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/batchmatmul_to_einsum.cc
LogicalResult matchAndRewrite(BatchMatMulOpType op, PatternRewriter& rewriter) const override { Value input_lhs = op.getX(); Value input_rhs = op.getY(); // LHS and RHS must be a ranked tensor type auto lhs_type = mlir::dyn_cast<RankedTensorType>(input_lhs.getType()); auto rhs_type = mlir::dyn_cast<RankedTensorType>(input_rhs.getType()); if (!lhs_type || !rhs_type) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.8K bytes - Viewed (0)