- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for input_list (0.17 sec)
-
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/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) -
tensorflow/compiler/mlir/lite/transforms/optimize_batch_matmul.cc
input, permutation_tensor_op.getResult()); }; Value input_lhs = bmm_op.getX(); Value input_rhs = bmm_op.getY(); Value output_lhs = bmm_op.getAdjX() ? create_z_x_transpose_op(input_lhs) : input_lhs; // The rhs need to be transposed if adj_y == false AND this matmul will be // legalized to tfl.fully_connected
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.6K bytes - Viewed (0)