Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 34 for ret_attrs (0.17 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc

                .getValue(),
            ftype);
    
        func->setAttrs(graph_func->getAttrs());
    
        llvm::SmallVector<Type> arg_types;
        llvm::SmallVector<Type> res_types;
        llvm::SmallVector<DictionaryAttr> arg_attrs;
        llvm::SmallVector<DictionaryAttr> res_attrs;
        if (failed(FilterTfgSpecificArgResultAttributes(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_uniform_attribute_utils.cc

      attrs.push_back(rewriter.getNamedAttr("output_quantization_axis",
                                            rewriter.getI64IntegerAttr(quant_dim)));
    
      op->setAttrs(rewriter.getDictionaryAttr(attrs));
    
      return success();
    }
    
    // This LogicalResult covers both the hybrid and fully quantized op cases.
    LogicalResult FillAttributesForUniformQuantizedConvolutionOp(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 18.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/utils/tftext_utils.cc

      func.eraseBody();
      func.addEntryBlock();
      func->setAttr(kTFImplements, attr);
      OpBuilder builder(func.getBody());
      std::string custom_option_buffer;
      if (failed(CreateNgramsCustomOption(func, attr.getAttrs(),
                                          custom_option_buffer))) {
        return failure();
      }
      auto op = builder.create<CustomOp>(
          func.getLoc(), func.getFunctionType().getResults(), func.getArguments(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_executor.cc

                 << ")";
        }
      }
      return success();
    }
    
    void GraphOp::print(OpAsmPrinter &p) {
      p << ' ';
      p.printRegion(getOperation()->getRegion(0));
      p.printOptionalAttrDict(getOperation()->getAttrs());
    }
    
    ParseResult GraphOp::parse(OpAsmParser &parser, OperationState &result) {
      llvm::SMLoc loc = parser.getCurrentLocation();
    
      // Parse the body region.
      Region &body = *result.addRegion();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 42.7K bytes
    - Viewed (0)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types.cc

        // OperationState so we can add regions to the new up.
        OperationState state(op->getLoc(), op->getName().getStringRef(), operands,
                             new_results, op->getAttrs(), op->getSuccessors());
        for (Region &region : op->getRegions()) {
          Region &new_region = *state.addRegion();
          rewriter.inlineRegionBefore(region, new_region, new_region.begin());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.6K bytes
    - Viewed (0)
Back to top