Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 18 for ret_attrs (0.58 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. tensorflow/compiler/mlir/lite/transforms/reduce_type_precision.cc

            op.getLoc(),
            /*result=*/
            mlir::cast<TensorType>(op.getResult().getType())
                .clone(builder.getI4Type()),
            /*operand=*/op.getOperands(), op->getAttrs());
        rewriter.replaceAllUsesWith(op.getResult(), new_gather_op.getResult());
    
        return success();
      }
    };
    
    class ReduceTypePrecisionPass
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.4K bytes
    - Viewed (0)
Back to top