- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for GetFunc (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/convert_launch_func_to_tf_call.cc
OpBuilder builder(launch); auto call_op = builder.create<TF::PartitionedCallOp>( module.getLoc(), launch.getResultTypes(), launch.getOperands(), SymbolRefAttr::get(builder.getContext(), launch.getFunc()), /*config=*/builder.getStringAttr(""), /*config_proto=*/builder.getStringAttr(""), /*executor_type=*/builder.getStringAttr(""));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 31 21:08:09 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/annotate_parameter_replication.cc
mirrored_replicate_args.insert( mlir::cast<IntegerAttr>(mirrored_index).getInt()); } } auto func = llvm::cast<func::FuncOp>(m.lookupSymbol(cluster_func.getFunc())); for (auto entry : llvm::enumerate(cluster_func.getOperands())) { auto operand = SkipIdentityAndReadVariable(entry.value()); auto block_arg = mlir::dyn_cast<BlockArgument>(operand);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
if (!status->status.ok()) return; auto run_metadata = context->ExportRunMetadata(); status->status = MessageToBuffer(*run_metadata, buf); } namespace { TFE_Op* GetFunc(TFE_Context* ctx, const tensorflow::NameAttrList& func, TF_Status* status) { TFE_Op* func_op = TFE_NewOp(ctx, func.name().data(), status); for (const auto& attr : func.attr()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 08:11:23 UTC 2024 - 44K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
// Look up function definition from module. mlir::func::FuncOp func = cluster_func->getParentOfType<ModuleOp>() .lookupSymbol<mlir::func::FuncOp>(cluster_func.getFunc()); bool use_spmd = false; if (auto use_spmd_attr = cluster_func->getAttrOfType<BoolAttr>(kUseSpmdAttr)) use_spmd = use_spmd_attr.getValue(); auto num_cores_per_replica_attr =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 28.9K bytes - Viewed (0) -
src/cmd/compile/internal/types/sym.go
func (sym *Sym) SetUniq(b bool) { sym.flags.set(symUniq, b) } func (sym *Sym) SetSiggen(b bool) { sym.flags.set(symSiggen, b) } func (sym *Sym) SetAsm(b bool) { sym.flags.set(symAsm, b) } func (sym *Sym) SetFunc(b bool) { sym.flags.set(symFunc, b) } func (sym *Sym) IsBlank() bool { return sym != nil && sym.Name == "_" } // Deprecated: This method should not be used directly. Instead, use a
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Thu Jan 26 21:56:56 UTC 2023 - 4.2K bytes - Viewed (0) -
src/cmd/compile/internal/ir/func.go
// // TODO(mdempsky): I suspect there's no need for separate fpos and // npos. func NewFunc(fpos, npos src.XPos, sym *types.Sym, typ *types.Type) *Func { name := NewNameAt(npos, sym, typ) name.Class = PFUNC sym.SetFunc(true) fn := &Func{Nname: name} fn.pos = fpos fn.op = ODCLFUNC // Most functions are ABIInternal. The importer or symabis // pass may override this. fn.ABI = obj.ABIInternal fn.SetTypecheck(1)
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Thu May 23 01:05:44 UTC 2024 - 21.1K bytes - Viewed (0)