- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 50 for TEST_F (0.37 sec)
-
tensorflow/cc/saved_model/saved_model_bundle_lite_test.cc
// when loading and unloading large numbers of SavedModelBundles. // TODO(sukritiramesh): Increase run iterations and move outside of the test // suite. TEST_F(LoaderTest, ResourceLeakTest) { SavedModelBundleLite bundle; SessionOptions session_options; RunOptions run_options; const string export_dir =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 14 19:16:58 UTC 2023 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config_test.cc
} return main; } protected: MLIRContext context_; OwningOpRef<ModuleOp> module_; }; TEST_F(LegalizationOpConfigTest, FailsWithExpectsLegalizationWithMlir) { TF_EXPECT_OK(CreateMlirModule()); EXPECT_FALSE(IsOpLegalizedWithMlir(*module_->getOperation())); } TEST_F(LegalizationOpConfigTest, ExpectsFalseForNonMlirOps) { TF_EXPECT_OK(CreateMlirModule());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 30 03:31:01 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/cc/gradients/linalg_grad_test.cc
EXPECT_LT(max_error, 1e-3); } Scope scope_; }; TEST_F(LinalgGradTest, Einsum_Transpose) { TensorShape x_shape({2, 3}); Output x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape)); auto y = Einsum(scope_, {x}, "ij->ji"); TensorShape y_shape({3, 2}); RunTest({x}, {x_shape}, {y}, {y_shape}); } TEST_F(LinalgGradTest, Einsum_TransposeBroadcast) { TensorShape x_shape({3, 2, 3});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 07 23:11:54 UTC 2022 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/func_test.cc
namespace mlir::quant { namespace { using ::testing::IsNull; using ::testing::NotNull; using FindMainFuncOpTest = ::mlir::quant::QuantizationTestBase; TEST_F(FindMainFuncOpTest, ReturnsMainFuncOp) { constexpr absl::string_view kModuleWithMainFunc = R"mlir( module { func.func @main() -> () { return } } )mlir";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 09:05:02 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/lower_cluster_to_runtime_ops_test.cc
Env* env_; std::string test_dir_; std::string test_group_name_; }; TEST_F(LowerClusterToRuntimeOpsTest, SanityCheck) { TF_ASSERT_OK(CreateMlirModule("empty_func.mlir")); TF_EXPECT_OK(RunLowerClusterToRuntimeOpsPassPipeline( *mlir_module_, DeviceType(DEVICE_TPU_XLA_JIT))); } TEST_F(LowerClusterToRuntimeOpsTest, LowersClusterOpsTPU) { TF_ASSERT_OK(CreateMlirModule("basic_cluster.mlir"));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:44:37 UTC 2024 - 6.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model_test.cc
const LogicalResult parse_result = parseSourceString(module_op_str, &block, ParserConfig(&ctx)); EXPECT_TRUE(succeeded(parse_result)); return cast<ModuleOp>(block.front()); } TEST_F(TfSavedModelTest, GetInitializerFunctionReturnsNullWhenNoSessionInitializerOp) { constexpr StringRef kModuleOpStr = R"mlir(module attributes {tf_saved_model.semantics} {})mlir"; Block block;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 05:03:09 UTC 2023 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/const_op_size_test.cc
EXPECT_TRUE(succeeded(parse_result)); auto const_op = dyn_cast_or_null<TF::ConstOp>(block.front()); EXPECT_TRUE(const_op); return const_op; } TEST_F(GetSizeInBytesTest, Int32ScalarConstOpSizeInBytes) { constexpr absl::string_view kConstOpExpr = R"mlir(%cst = "tf.Const"() {value = dense<1> : tensor<i32>} : () -> tensor<i32>)mlir"; Block block{};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 14 04:37:13 UTC 2022 - 5.4K bytes - Viewed (0) -
tensorflow/cc/saved_model/reader_test.cc
"tensorflow/serving/predict"); } }; TEST_F(ReaderTest, TagMatch) { MetaGraphDef meta_graph_def; const string export_dir = GetDataDependencyFilepath(TestDataSharded()); TF_ASSERT_OK(ReadMetaGraphDefFromSavedModel(export_dir, {kSavedModelTagServe}, &meta_graph_def)); CheckMetaGraphDef(meta_graph_def); } TEST_F(ReaderTest, NoTagMatch) { MetaGraphDef meta_graph_def;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 17 21:17:08 UTC 2023 - 5.1K bytes - Viewed (0) -
tensorflow/cc/ops/while_loop_test.cc
// TODO(skyewm): test bad cond output shape TEST_F(WhileLoopTest, NullCondOutputNode) { Init(1); // TODO(skyewm): improve error message CreateLoop( [](const Scope& s, const std::vector<Output>& inputs, Output* output) { *output = {nullptr, 0}; return s.status(); }, AddOneBody, error::INVALID_ARGUMENT, "Node is null"); } TEST_F(WhileLoopTest, InvalidCondOutputIndex) { Init(1);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 13 22:30:58 UTC 2023 - 6.4K bytes - Viewed (0) -
tensorflow/cc/saved_model/util_test.cc
} } SignatureDef sig_; google::protobuf::Map<std::string, TensorProto> request_; std::map<std::string, TensorProto> unaliased_request_; TensorProto input_x_, input_y_, default_x_, default_y_; }; TEST_F(GetInputValuesTest, RequestContainsInvalidInputs) { google::protobuf::Map<std::string, TensorProto> local_request = request_; local_request["xx"] = CreateTensorProto(2); std::vector<std::pair<string, Tensor>> inputs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 5.3K bytes - Viewed (0)