Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 21 for input_list (0.12 sec)

  1. 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)
  2. 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)
  3. 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)
  4. tensorflow/compiler/mlir/lite/experimental/tac/transforms/pick_subgraphs.cc

    //       total_cost = 0
    //       for input_subgraph in current_subgraph.input_subgraphs:
    //         input_cost = kInfinity
    //         for input_target in input_subgraphs.upported_targets:
    //           # cost = aggregated cost for input_subgraph with transfer cost.
    //           input_cost = min(input_cost, cost)
    //         total_cost += input_cost
    //       total_cost += current_subgraph.get_computation_cost(target)
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 24 15:10:02 UTC 2022
    - 19.7K bytes
    - Viewed (0)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top