- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for created (0.09 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc
} else { return false; } return true; } // Create a dummy zero to be fed locally from the host to the TPUExecute. Value CreateZeroInput(Location loc, OpBuilder& builder, Attribute zero_attr, DenseIntElementsAttr shape_attr) { ConstOp zero = builder.create<ConstOp>(loc, zero_attr); zero->setAttr(kICIWeightDistributionMlirBridgeMarker,
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/mlir_graph_optimization_pass.cc
// be created). static void DumpModule(mlir::ModuleOp module, std::string file_prefix) { std::string prefix = GetDumpDirFromEnvVar(); if (prefix.empty()) return; auto* env = tensorflow::Env::Default(); auto status = env->RecursivelyCreateDir(prefix); if (!status.ok()) { LOG(WARNING) << "cannot create directory '" << prefix << "': " << status.message();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 18.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.cc
for (const auto& device : devices) if (DeviceNameUtils::IsCompleteSpecification(spec, device)) matching_devices.push_back(device); return matching_devices; } // Create error message for a conflicting attribute of a device. template <typename T> absl::Status MismatchedTPUSystemAttributeErr(absl::string_view attribute, T a, T b) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:10:40 UTC 2024 - 32.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf_test.cc
EXPECT_EQ(compilation_status.Delta(kOldBridgeWithFallbackModeSuccess), 0); } TEST(LegalizeTFTest, RecordsStreamzForNoMlirFallback) { FunctionDef my_func = tensorflow::FunctionDefHelper::Create("empty", {}, {}, {}, {}, {}); tensorflow::FunctionDefLibrary fdef; *(fdef.add_function()) = my_func; tensorflow::FunctionLibraryDefinition flib_def( tensorflow::OpRegistry::Global(), fdef);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
nodes_[inst] = node; return absl::OkStatus(); } bool IsEntryFunctionArg(BlockArgument arg) { return arg.getParentRegion()->getParentOfType<FuncOp>().getName() == "main"; } // Creates argument nodes from Block argument. If a name is supplied, that // name will be used instead of generating a unique name. Status Exporter::AddArgumentNode(BlockArgument arg, unsigned index,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor.cc
TF_RETURN_IF_ERROR(c_event->Create()); return std::move(c_event); } absl::StatusOr<std::unique_ptr<Stream>> CreateStream( std::optional<std::variant<StreamPriority, int>> priority = std::nullopt) override { auto stream = std::make_unique<CStream>(&device_, stream_executor_, this); TF_RETURN_IF_ERROR(stream->Create()); return std::move(stream); } private:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 14 07:39:19 UTC 2024 - 27.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_passes.cc
#include "tensorflow/compiler/mlir/tensorflow/transforms/passes.h" #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_passes.h" #include "xla/mlir_hlo/mhlo/transforms/passes.h" namespace mlir { /// Create a pass to convert from the TFExecutor to the TF control dialect. std::unique_ptr<OperationPass<func::FuncOp>> CreateTFExecutorToControlDialectConversion(); } // namespace mlir namespace tensorflow { namespace {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 18:45:51 UTC 2024 - 25.5K bytes - Viewed (0)