Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 19 of 19 for input_list (0.17 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
Back to top