- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for input_list (0.16 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/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/quantization/stablehlo/tests/bridge/convert-tf-quant-types.mlir
// CHECK: %[[input:.*]] = "tf.ConcatV2"(%arg0, %arg1, %[[VAL:.*]]) : (tensor<3x3xi8>, tensor<3x3xi8>, tensor<i64>) -> tensor<6x3xi8> // CHECK: %[[input_qint:.*]] = "tf.Cast"(%[[input]]) <{Truncate = false}> : (tensor<6x3xi8>) -> tensor<6x3x!tf_type.qint8> // CHECK: %[[output:.*]] = "tf.UniformDequantize"(%[[input_qint]] // CHECK: return %[[output]] : tensor<6x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 25.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/unroll_batch_matmul.cc
template <typename BatchMatMulOpType> LogicalResult ConvertTFBatchMatMulOp<BatchMatMulOpType>::matchAndRewrite( BatchMatMulOpType op, PatternRewriter& rewriter) const { Value input_lhs = op.getX(); Value input_rhs = op.getY(); if (!mlir::isa<RankedTensorType>(input_lhs.getType())) { // LHS must be a ranked tensor type return failure(); } if (!mlir::isa<RankedTensorType>(input_rhs.getType())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
if (auto input_result = mlir::dyn_cast<mlir::OpResult>(src)) { auto* input_inst = GetIslandInnerOpOrSelf(input_result.getOwner()); // Replaces the input node with NextIteration sink if it is a NextIteration // source. if (auto next_iter_source = llvm::dyn_cast<mlir::tf_executor::NextIterationSourceOp>( input_inst)) input_inst = next_iter_source.GetSink();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
if (auto input_result = mlir::dyn_cast<mlir::OpResult>(src)) { auto* input_inst = GetIslandInnerOpOrSelf(input_result.getOwner()); // Replaces the input node with NextIteration sink if it is a NextIteration // source. if (auto next_iter_source = llvm::dyn_cast<mlir::tf_executor::NextIterationSourceOp>( input_inst)) input_inst = next_iter_source.GetSink();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/cc/framework/fuzzing/cc_op_fuzz_gen.cc
"_1 = ", "tensorflow::ops::Placeholder(scope.WithOpName(\"", arg.name(), "\"), ", type, ");\n"); strings::StrAppend( out, absl::Substitute(" InputList $0({$0_0, $0_1});\n", arg.name())); } else { strings::StrAppend( out, " auto ", arg.name(), " = ",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jan 27 16:26:51 UTC 2024 - 13K bytes - Viewed (0)