- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for AddAttributes (0.23 sec)
-
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) -
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) -
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) -
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) -
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)