Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 37 for get_attr (0.16 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/optimize.cc

    TF::ConstOp GetI64ConstantTensor(PatternRewriter &rewriter,
                                     ArrayRef<int64_t> values, Location location) {
      auto cst_attr = rewriter.getI64TensorAttr(values);
      return rewriter.create<TF::ConstOp>(location, cst_attr.getType(), cst_attr);
    }
    
    // Rewrites broadcast->reshape to a reshape->broadcast that reduces
    // the rank of the input and output of the broadcast.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.1K bytes
    - Viewed (0)
  2. src/cmd/vendor/golang.org/x/sys/unix/zsymaddr_zos_s390x.s

    	MOVD $·Fremovexattr(SB), R8
    	MOVD R8, ret+0(FP)
    	RET
    
    // THIS FILE IS GENERATED BY THE COMMAND AT THE TOP; DO NOT EDIT
    
    TEXT ·get_FgetxattrAddr(SB), NOSPLIT|NOFRAME, $0-8
    	MOVD $·Fgetxattr(SB), R8
    	MOVD R8, ret+0(FP)
    	RET
    
    // THIS FILE IS GENERATED BY THE COMMAND AT THE TOP; DO NOT EDIT
    
    TEXT ·get_FsetxattrAddr(SB), NOSPLIT|NOFRAME, $0-8
    	MOVD $·Fsetxattr(SB), R8
    	MOVD R8, ret+0(FP)
    	RET
    
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Wed May 08 16:12:58 UTC 2024
    - 8.8K bytes
    - Viewed (0)
  3. 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)
  4. tensorflow/compiler/mlir/tensorflow/transforms/xla_rewrite.cc

        }
      }
    
      if (!in_order) {
        // Functions do not get reused in practice, so skip the check for if the
        // callee has been updated.
        StringAttr callee_sym = cluster_func_op.getFuncAttr().getAttr();
        MoveResourceArgsToEnd(symtab.lookup<func::FuncOp>(callee_sym));
      }
      builder.setInsertionPoint(cluster_func_op);
      auto xla_launch_op = builder.create<TF::XlaLaunchOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 4.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/legalize_variables.cc

      void runOnOperation() override {
        auto module = getOperation();
        // If TFLite variable legalization is not allowed, then we skip this pass.
        if (auto legalize_tfl_variables_attr =
                module->getAttr(kLegalizeTflVariables)) {
          if (!mlir::cast<BoolAttr>(legalize_tfl_variables_attr).getValue()) return;
        }
    
        RewritePatternSet patterns(&getContext());
        populateWithGenerated(patterns);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 3.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_custom_call_to_composite.cc

      LogicalResult matchAndRewrite(mlir::stablehlo::CustomCallOp op,
                                    PatternRewriter &rewriter) const override {
        auto backendConfig =
            mlir::dyn_cast<DictionaryAttr>(op->getAttr("composite.backend_config"));
        if (!backendConfig)
          return op->emitError(
              "custom_call has no 'composite.backend_config' attribute or the "
              "attribute is not a dictionary");
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 4.4K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/cluster_outlining.cc

      cluster_op->setAttr(
          builder->getStringAttr(kFuncAttr),
          mlir::SymbolRefAttr::get(builder->getContext(), outlined_func.getName()));
    
      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());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  8. 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)
  9. 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)
  10. 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)
Back to top