- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 1,512 for q_builder (0.18 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc
push.replaceAllUsesWith(push.getElem()); OpBuilder builder(push); // Read the current buffer and size. auto stack_val = cutil::ReadLocalVariable(push.getHandle(), builder, push.getLoc()); auto index = cutil::ReadLocalVariable(it->getSecond(), builder, push.getLoc()); stack_val = cutil::SetElement(index, stack_val, push.getElem(), builder, push.getLoc());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc
} } return WalkResult::advance(); }); OpBuilder builder(replicate); OpBuilder inner_builder = OpBuilder::atBlockBegin(&cluster.getBody().front()); // mapping from original operand to new XlaAllReduce op. llvm::DenseMap<Value, Value> orig_to_new; for (Value bcast : bcasts) { if (failed(MoveBroadcastToCluster(builder, inner_builder, cluster,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/get_alternative_subgraph.cc
OpBuilder* builder) { func::FuncOp cloned_func = func.clone(); cloned_func.setPrivate(); auto interface_name = GetInterFaceName(func); if (!interface_name.has_value()) { func.emitError("the func op does not have interface_name"); return nullptr; } cloned_func->setAttr( kDevice, builder->getStringAttr(target_device_inference_type.hardware));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
} } // A helper class to inline TF::StatefulPartitionedCall ops struct Inliner : public InlinerInterface { Inliner(OpBuilder& builder, SymbolTable& symbol_table) : InlinerInterface(builder.getContext()), builder(builder), symbol_table(symbol_table) {} bool isLegalToInline(Operation* call, Operation* callable, bool wouldBeCloned) const override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
} Operation* ApplyXlaHostTransferAttr(Operation* op, OpBuilder& builder) { op->setAttr("_xla_has_host_transfer", builder.getBoolAttr(true)); return op; } // Creates a tf._XlaSendFromHost or tf._XlaSendFromHostV2 op. If device ordinal // is present, a tf._XlaSendFromHostV2 op is created instead. Operation* CreateSendFromHostOp(OpBuilder& builder, Location loc,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
src/internal/types/testdata/fixedbugs/issue45550.go
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Tue Jan 17 19:54:25 UTC 2023 - 314 bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/windows/BUILD
tags = [ "manual", "nobuilder", "notap", ], visibility = ["//visibility:public"], deps = [":windows_filesystem_impl"], ) # The real implementation of the filesystem. cc_library( name = "windows_filesystem_impl", srcs = ["windows_filesystem.cc"], copts = get_win_copts(), tags = [ "manual", "nobuilder", "notap", ],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 20 06:38:26 UTC 2024 - 936 bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/raise_target_subgraphs.cc
void RaiseTargetSubgraphsForBlock( Block& block, OpBuilder& builder, ModuleOp module, bool skip_cpu, int& func_count, const TF::SideEffectAnalysis::Info& side_effect_info); }; // After raising ops and adding the Func & Call op, call this function // to set attributes specific to this pass. void AddAttrs(OpsAdded& ops_added, OpBuilder& builder, int func_count) { func::FuncOp& added_func_op = ops_added.func_op;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_patterns.td
CPred<"$_self.isa<DenseElementsAttr>()">, "non-opaque constant tensor">; def CreateGatherNdOp : NativeCodeCall< "$_builder.create<TF::GatherNdOp>($0.getLoc(), $0.getType(), $1, $2)">; def CreateTFCastOpI32 : NativeCodeCall< "CreateTFCastOpI32(&$_builder, $_loc, $0, $1)">; // Convert to std constant for statically shaped, non-opaque constants.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/c/ops.cc
TF_Status* status) { auto* cc_builder = reinterpret_cast<OpDefBuilder*>(builder); TF_SetStatus(status, TF_OK, ""); ::tensorflow::OpRegistry::Global()->Register( [cc_builder](::tensorflow::OpRegistrationData* op_reg_data) -> Status { Status result = cc_builder->Finalize(op_reg_data); delete cc_builder; return result; }); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 28 22:41:35 UTC 2022 - 10.9K bytes - Viewed (0)