Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for AddAttributes (0.23 sec)

  1. tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc

      const std::string tf_op_full_name = absl::StrCat("tf.", node_def.op());
      mlir::OperationState op_state(loc, tf_op_full_name);
      op_state.addOperands(func.getArguments());
      op_state.addTypes(output_tys);
      op_state.addAttributes(attrs);
      mlir::Operation* tf_op = op_builder.create(op_state);
      op_builder.create<mlir::func::ReturnOp>(loc, tf_op->getResults());
    
      // Run the decompose passes on the module
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 29 02:34:43 UTC 2024
    - 9.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc

        llvm::SmallVector<mlir::Type, 4> output_tys;
        for (int i = 0; i < custom_op.getNumResults(); i++) {
          output_tys.push_back(custom_op.getType(i));
        }
        op_state.addTypes(output_tys);
        op_state.addAttributes(attr);
        auto stablehlo_op = builder.create(op_state);
        custom_op.replaceAllUsesWith(stablehlo_op);
        custom_op.erase();
      });
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jan 24 06:08:43 UTC 2024
    - 10.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc

                          << ". Please make sure the op library for "
                          << parsed_op_name << " is linked properly";
          return failure();
        }
        op_state.addAttributes(attrs);
    
        Operation* tf_op = rewriter.create(op_state);
        rewriter.replaceOp(op, tf_op->getResults());
    
        // Special type fixes for TF Resource Tensors that are casted to
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

        }
    
        attrs.emplace_back(builder.getNamedAttr(
            "decomposition",
            mlir::vhlo::StringV1Attr::get(builder.getContext(), decomposition)));
      }
    
      op_state.addAttributes(attrs);
    
      // Handle the conversion from subgraph index to functions for If and While. We
      // will add CallOps in the region to call the functions later for While.
      TF_ASSIGN_OR_RETURN(auto function_ref_attrs,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

      return nullptr;
    }
    
    void TFLDialect::initialize() {
      addOperations<
    #define GET_OP_LIST
    #include "tensorflow/compiler/mlir/lite/ir/tfl_ops.cc.inc"
          >();
      addAttributes<
    #define GET_ATTRDEF_LIST
    #include "tensorflow/compiler/mlir/lite/ir/tfl_ops_attrdefs.cc.inc"
          >();
      addInterfaces<TensorFlowLiteInlinerInterface,
                    TensorFlowLiteDialectFoldInterface>();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
Back to top