- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for StrEq (0.23 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/cc/graph_def_test.cc
#include "tsl/platform/protobuf.h" // IWYU pragma: keep namespace stablehlo::quantization { namespace { using ::tensorflow::GraphDef; using ::tensorflow::NodeDef; using ::testing::SizeIs; using ::testing::StrEq; using ::tsl::protobuf::TextFormat; TEST(GraphDefTest, MutateNodeDefsMutatesTopLevelNodeDefs) { GraphDef graph_def; ASSERT_TRUE(TextFormat::ParseFromString(R"pb(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 28 18:38:06 UTC 2023 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export_test.cc
ASSERT_THAT(exported_model.graph_def().node(), SizeIs(1)); EXPECT_THAT(exported_model.graph_def().node()[0].name(), StrEq("foo")); EXPECT_THAT(exported_model.init_node_name(), StrEq("init_node_name")); EXPECT_THAT(exported_model.checkpoint_dir(), StrEq("checkpoint_dir")); EXPECT_FALSE(exported_model.has_saver_def()); EXPECT_THAT(exported_model.function_aliases(), IsEmpty());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 19.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/calibration/representative_dataset_test.cc
ASSERT_THAT(*representative_dataset_file_map, SizeIs(1)); EXPECT_THAT(*representative_dataset_file_map, Contains(Key(StrEq("test_signature_key")))); EXPECT_THAT(representative_dataset_file_map->at("test_signature_key") .tfrecord_file_path(), StrEq("test_path")); } TEST(CreateRepresentativeDatasetFileMapTest, ConfigWithDuplicateSignatureKeyReturnsInvalidArgumentError) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jan 29 04:53:21 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/config_test.cc
EXPECT_THAT(default_spec.matcher().function_name().regex(), StrEq(".*")); EXPECT_TRUE(default_spec.method().has_static_range_ptq()); // Test that the expansion for convolution ops is done. const QuantizationSpec& conv_spec = new_config.specs().specs(1); EXPECT_THAT(conv_spec.matcher().function_name().regex(), StrEq("composite_conv.*")); ASSERT_TRUE(conv_spec.method().has_static_range_ptq());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 06:59:34 UTC 2024 - 12K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/pre_calibration_test.cc
using ::stablehlo::quantization::PopulateDefaults; using ::stablehlo::quantization::QuantizationConfig; using ::testing::Contains; using ::testing::SizeIs; using ::testing::StartsWith; using ::testing::StrEq; using ::tsl::testing::IsOk; // Matches an operation whose `getSymName` equals `name`. MATCHER_P(HasSymName, name, "") { auto non_const_arg = const_cast<std::remove_const_t<decltype(arg)>>(arg);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 21:41:08 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/report_test.cc
StrEq("composite_dot_general_fn_2")); EXPECT_TRUE(quantized_result.method().has_static_range_ptq()); // Test that the non-quantized op is captured in `results`. const QuantizationResult& non_quantized_result = results.results(1); EXPECT_THAT(non_quantized_result.quantizable_unit().name(), StrEq("composite_dot_general_fn_1"));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 10:10:34 UTC 2024 - 18.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/instrumentations/save_report_test.cc
namespace mlir::quant::stablehlo { namespace { using ::stablehlo::quantization::QuantizationResults; using ::stablehlo::quantization::io::ReadFileToString; using ::testing::SizeIs; using ::testing::StrEq; using ::tsl::protobuf::TextFormat; using ::tsl::testing::IsOk; using ::tsl::testing::StatusIs; using SaveQuantizationReportInstrumentationTest = QuantizationTestBase;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 02:59:01 UTC 2024 - 9.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call_test.cc
SmallVector<func::FuncOp> funcs = GetSortedFunctions(*module_op); ASSERT_THAT(funcs, SizeIs(3)); EXPECT_THAT(funcs[0].getSymName(), StrEq("conv_1_fn")); EXPECT_THAT(funcs[1].getSymName(), StrEq("conv_2_fn")); EXPECT_THAT(funcs[2].getSymName(), StrEq("conv_3_fn")); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 26.2K bytes - Viewed (0)