- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 20 for compilation_status (0.3 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/embedding_program_key.mlir
// CHECK: TPUCompileMlir // CHECK: "tf.OpA"(%[[COMPILE_LAUNCH]]#1 %0:2 = "tf_device.launch"() ({ %compilation_status, %program = "tf._TPUCompileMlir"() { metadata = "...", mlir_module = "..." } : () -> (tensor<!tf_type.string>, tensor<3x!tf_type.string>) tf_device.return %compilation_status, %program : tensor<!tf_type.string>, tensor<3x!tf_type.string>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 14:28:22 UTC 2023 - 16.4K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compilation_cache.h
} { mutex_lock lock(entry->mu); if (compile_state.has_value()) { entry->compile_state = *compile_state; } if (compilation_status.has_value()) { entry->compilation_status = *compilation_status; } if (compilation_result.has_value()) { entry->compilation_result = std::move(*compilation_result); } if (executable.has_value()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 12 08:49:52 UTC 2023 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_dialect_to_executor_test.cc
CellReader<int64_t> compilation_status(kExportStreamzName); TF_ASSERT_OK(CreateMlirModule("invalid_executor.mlir")); EXPECT_FALSE(ExportFromTensorflowDialectToExecutor(*mlir_module_).ok()); EXPECT_EQ(compilation_status.Delta(kExportSuccess), 0); EXPECT_EQ(compilation_status.Delta(kExportFailed), 1); } TEST_F(TensorflowDialectToExecutorTest, PrunesDeadOps) { CellReader<int64_t> compilation_status(kExportStreamzName);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 13 23:22:50 UTC 2024 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf_test.cc
EXPECT_EQ(compilation_status.Delta(kMlirCombinedOldSuccess), 1); EXPECT_EQ(compilation_status.Delta(kMlirCombinedOldFailure), 0); // Old bridge should never be called at all. EXPECT_EQ(compilation_status.Delta(kOldBridgeMlirFilteredFailure), 0); EXPECT_EQ(compilation_status.Delta(kOldBridgeWithFallbackModeFailure), 0); EXPECT_EQ(compilation_status.Delta(kOldBridgeMlirFilteredSuccess), 0);
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/tfrt/tests/fuse_tpu_compile_and_execute_ops.mlir
%2 = "tf.Shape"(%0) {device = "/CPU:0"} : (tensor<*xi32>) -> tensor<?xi64> %compilation_status, %program = "tf._TPUCompileMlir"(%1, %2) {device = "/CPU:0", metadata = "metadata", mlir_module = "mlir_module"} : (tensor<?xi64>, tensor<?xi64>) -> (tensor<!tf_type.string>, tensor<3x!tf_type.string>) "tf.TPUCompileSucceededAssert"(%compilation_status) {device = "/CPU:0"} : (tensor<!tf_type.string>) -> ()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 13.8K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compilation_cache_test.cc
EXPECT_EQ(cache_value_1->compilation_status.code(), error::INVALID_ARGUMENT); EXPECT_TRUE(cache_value_1->compilation_result != nullptr); EXPECT_TRUE(cache_value_1->executable != nullptr); EXPECT_EQ(cache_value_1->executable->data, "foo_exe"); EXPECT_EQ(cache_value_2->compile_state, DeviceCompileState::kCompiling); EXPECT_TRUE(cache_value_2->compilation_status.ok());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/cluster_tf_test.cc
CellReader<int64_t> compilation_status(kCompilationStreamz); TF_ASSERT_OK(CreateMlirModule("empty_func.mlir")); TF_EXPECT_OK(RunFunctionTf2xlaClusteringBridge( *mlir_module_, /*is_supported_by_replicated_brige*/ true, /*is_in_fallback_enabled_mode=*/true)); EXPECT_EQ(compilation_status.Delta(mlir::TF::kMlirPh1BridgeCounterReplicated, mlir::TF::kMlirPh1BridgeCounterV2,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:44:37 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/cluster_tf_test.cc
}; TEST_F(SessionClusterTensorflowDialectTest, ClustersTf) { CellReader<int64_t> compilation_status(kCompilationStreamz); TF_ASSERT_OK(CreateMlirModule("empty_func.mlir")); TF_EXPECT_OK( RunSessionTf2xlaClusteringBridge(*mlir_module_, /*is_in_fallback_enabled_mode=*/false)); EXPECT_EQ(compilation_status.Delta(mlir::TF::kMlirPh1BridgeCounterReplicated,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:44:37 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/tf_dialect_to_executor_test.cc
CellReader<int64_t> compilation_status(kCompilationStreamz); TF_ASSERT_OK(CreateMlirModule("empty_func.mlir")); TF_EXPECT_OK(ExportFromTensorflowDialectToExecutor(*mlir_module_)); EXPECT_EQ(compilation_status.Delta("success"), 1); } TEST_F(TensorflowDialectToExecutorTest, ErrorsWhenCannotConvert) { CellReader<int64_t> compilation_status(kCompilationStreamz);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 08:08:57 UTC 2024 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph_test.cc
absl::Status compilation_status = tensorflow::tf2xla::v1::CompileTensorflowGraphToHlo( computation, metadata_proto, use_tuple_args, shape_determination_fns, arg_shapes, &arg_core_mapping, &per_core_arg_shapes, client, &compilation_result); if (!compilation_status.ok()) return compilation_status; return compilation_result; } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 08:08:57 UTC 2024 - 11.7K bytes - Viewed (0)