- Sort Score
- Result 10 results
- Languages All
Results 11 - 19 of 19 for input_list (0.17 sec)
-
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/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) -
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) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
}; if (!can_convert_to_fully_connected(op)) { return false; } Value input_lhs = get_real_input_value(op.getX()); Value input_rhs = get_real_input_value(op.getY()); Value legalized_lhs = 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: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/cc/framework/cc_op_gen_util.cc
const auto& api_def_arg = *FindInputArg(api_def.arg_order(i), api_def); arg_types.push_back(strings::StrCat( "::tensorflow::", ArgIsList(arg) ? "InputList" : "Input")); arg_names.push_back(AvoidCPPKeywords(api_def_arg.rename_to())); // TODO(keveman): Include input type information. StringPiece description = api_def_arg.description();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 00:57:05 UTC 2024 - 25K bytes - Viewed (0) -
pkg/controller/podautoscaler/horizontal.go
// it is not present. The new list will be returned. func setConditionInList(inputList []autoscalingv2.HorizontalPodAutoscalerCondition, conditionType autoscalingv2.HorizontalPodAutoscalerConditionType, status v1.ConditionStatus, reason, message string, args ...interface{}) []autoscalingv2.HorizontalPodAutoscalerCondition { resList := inputList var existingCond *autoscalingv2.HorizontalPodAutoscalerCondition
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Sat May 04 18:33:12 UTC 2024 - 63.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
arg_nodes->resize(inputs.size()); ret_nodes->resize(outputs.size()); for (Node* n : GetOrderedNodes()) { // Handle inputs/arguments. auto input_it = inputs.find(n->name()); if (input_it != inputs.end()) { (*arg_nodes)[input_it->second] = {n, 0}; } // Handle outputs/returns. if (output_node_names.contains(n->name())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)