Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 68 for ret_attrs (1.48 sec)

  1. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

      // construction.
      llvm::SmallVector<mlir::NamedAttrList, 4> arg_attrs;
      arg_attrs.resize(func.getNumArguments());
      llvm::SmallVector<mlir::NamedAttrList, 4> ret_attrs;
      ret_attrs.resize(func.getNumResults());
    
      auto set_attributes_on_func = [&](Node* node, int64_t index, bool is_arg) {
        for (const auto& node_attr : node->attrs()) {
          const auto& key = node_attr.first;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  2. 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)
  3. 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)
  4. tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc

        auto& info = resource_it.second;
    
        if (info.is_read) {
          Operation* read = builder.create<TF::ReadVariableOp>(
              op_->getLoc(), info.data_type, resource);
          read->setAttrs(info.read_attrs ? info.read_attrs : empty_attrs);
          read->removeAttr(kDeviceAttr);
          info.hoisted_read = read->getResult(0);
        }
      }
    }
    
    // Replaces all resource reads with the hoisted read.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 55.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc

        nested_module->emitOpError("unexpected already present outlined module.");
        return signalPassFailure();
      }
      ModuleOp outlined_module = ModuleOp::create(getOperation().getLoc());
      outlined_module->setAttrs(getOperation()->getAttrDictionary());
      outlined_module->setAttr(SymbolTable::getSymbolAttrName(),
                               StringAttr::get(ctx, kNestedModule));
      symbol_table.insert(outlined_module);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.7K bytes
    - Viewed (0)
  6. staging/src/k8s.io/apiserver/pkg/storage/selection_predicate.go

    	Field               fields.Selector
    	GetAttrs            AttrFunc
    	IndexLabels         []string
    	IndexFields         []string
    	Limit               int64
    	Continue            string
    	AllowWatchBookmarks bool
    }
    
    // Matches returns true if the given object's labels and fields (as
    // returned by s.GetAttrs) match s.Label and s.Field. An error is
    // returned if s.GetAttrs fails.
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Tue Jun 04 09:20:10 UTC 2024
    - 5.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/rewrite_tpu_embedding_ops.cc

      auto operands = llvm::to_vector<4>(op->getOperands());
      operands.push_back(operand);
      auto new_op = builder->create<OpT>(op->getLoc(), op->getResultTypes(),
                                         operands, attr.getAttrs());
      op->replaceAllUsesWith(new_op.getOperation()->getResults());
      op->erase();
      return new_op;
    }
    
    // Returns success if the function has at most one op of the template type and
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 22:55:42 UTC 2024
    - 4.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/stablehlo/odml_converter/folders.cc

          return failure();
        }
        res.push_back(signed_lhs_val / signed_rhs_val);
      }
    
      auto res_attr = DenseElementsAttr::get(
          const_oprs[0].getType().cast<RankedTensorType>(), res);
      rewriter.replaceOpWithNewOp<stablehlo::ConstantOp>(adaptor.value().Op(),
                                                         res_attr);
      return success();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 06:11:55 UTC 2024
    - 4.5K bytes
    - Viewed (0)
  9. 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)
  10. 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)
Back to top