Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 68 for ret_attrs (0.22 sec)

  1. tensorflow/compiler/mlir/tfr/ir/tfr_ops.td

      }];
    
      let arguments = (ins
        TypeAttrOf<FunctionType>:$function_type,
        StrAttr:$sym_name,
        OptionalAttr<DictArrayAttr>:$arg_attrs,
        OptionalAttr<DictArrayAttr>:$res_attrs
      );
    
      let results = (outs);
    
      // When the regions is empty, the tfr.func is an external function and used
      // to model the element type constraints of the tf op. Otherwise, there is one
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 10:54:29 UTC 2024
    - 17.4K bytes
    - Viewed (0)
  2. pkg/registry/core/service/strategy.go

    func (serviceStatusStrategy) WarningsOnUpdate(ctx context.Context, obj, old runtime.Object) []string {
    	return nil
    }
    
    // GetAttrs returns labels and fields of a given object for filtering purposes.
    func GetAttrs(obj runtime.Object) (labels.Set, fields.Set, error) {
    	service, ok := obj.(*api.Service)
    	if !ok {
    		return nil, nil, fmt.Errorf("not a service")
    	}
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Thu Apr 11 13:09:36 UTC 2024
    - 14.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/cluster_outlining.cc

      builder->setInsertionPoint(cluster_op);
      auto cluster_func_op = builder->create<tf_device::ClusterFuncOp>(
          cluster_op.getLoc(), outlined_func.getFunctionType().getResults(),
          live_ins.getArrayRef(), cluster_op->getAttrs());
      cluster_op.replaceAllUsesWith(cluster_func_op);
      cluster_op.erase();
    }
    
    // Outlines body of `tf_device.launch` into a function and create a
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/utils/nms_utils.cc

                     StringAttr::get(func_.getContext(), kCustomSSDPostprocessing));
    
      OpBuilder builder(func_.getBody());
      std::string custom_option_buffer;
      if (failed(CreateNMSCustomOptions(func_, attr_.getAttrs(),
                                        custom_option_buffer))) {
        return failure();
      }
      auto op = builder.create<CustomOp>(
          func_.getLoc(), func_.getFunctionType().getResults(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_arguments.cc

        }
      }
      OpBuilder builder(op);
      builder.setInsertionPointAfter(op);
      OperationState state(op->getLoc(), op->getName().getStringRef(),
                           op->getOperands(), result_types, op->getAttrs());
      for (int i = 0; i < op->getNumRegions(); ++i) {
        state.addRegion();
      }
      Operation* new_op = builder.create(state);
      for (const auto& indexed_regions : llvm::enumerate(op->getRegions())) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/utils/call_graph_util.cc

        }
      });
      return entry_funcs;
    }
    
    LogicalResult GetCallees(SymbolUserOpInterface op, SymbolTable &symtab,
                             llvm::SmallVector<func::FuncOp> &callees) {
      for (auto attr : op->getAttrs()) {
        auto sym = mlir::dyn_cast<SymbolRefAttr>(attr.getValue());
        if (!sym) continue;
        auto callee = symtab.lookup<func::FuncOp>(sym.getRootReference());
        if (!callee) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 2.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc

        return func_.emitWarning()
               << "Invalid number of results from " << kMaxUnpooling << ": "
               << func_.getFunctionType().getNumResults();
      }
    
      auto attrs = attr_.getAttrs();
    
      if (failed(HasIntegerArrayWithSize(&func_, attrs, "pool_size", 2))) {
        return failure();
      }
    
      if (failed(HasIntegerArrayWithSize(&func_, attrs, "strides", 2))) {
        return failure();
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/transforms/push_transpose_through_ewise.cc

        auto *new_ewise_op =
            rewriter.create(op->getLoc(), op->getName().getIdentifier(),
                            {tpose_arg1.getOperand(0), tpose_arg2.getOperand(0)},
                            new_out_type, op->getAttrs());
    
        // Apply original tranpose to output of ewise op.
        auto out_tpose_op = rewriter.create<TFL::TransposeOp>(
            new_ewise_op->getLoc(), op->getResult(0).getType(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.5K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

        new_while_operands.push_back(it->getSecond().size);
      }
      auto new_while = builder.create<TF::WhileOp>(
          while_op.getLoc(), body.getFunctionType().getInputs(), new_while_operands,
          while_op->getAttrs());
      for (const auto& entry : output_buffer_to_size) {
        (*buffer_to_size)[new_while.getResult(std::get<0>(entry))] = {
            new_while.getResult(std::get<1>(entry)), std::get<2>(entry)};
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc

      for (const auto& attr_def : op_reg_data->op_def.attr()) {
        registered_attrs.insert(attr_def.name());
      }
      // Attributes are not in the registered attributes set will be ignored.
      for (auto& attr : inst->getAttrs()) {
        if (registered_attrs.find(attr.getName()) == registered_attrs.end()) {
          attrs_to_ignore->insert(
              absl::string_view(attr.getName().data(), attr.getName().size()));
        }
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 11.1K bytes
    - Viewed (0)
Back to top