- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 52 for TEST_F (0.12 sec)
-
tensorflow/c/experimental/stream_executor/stream_executor_test.cc
} SP_Platform platform_; SP_PlatformFns platform_fns_; SP_DeviceFns device_fns_; SP_StreamExecutor se_; SP_TimerFns timer_fns_; std::unique_ptr<CPlatform> cplatform_; }; TEST_F(StreamExecutorTest, Allocate) { se_.allocate = [](const SP_Device* const device, uint64_t size, int64_t memory_space, SP_DeviceMemoryBase* const mem) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 19:54:04 UTC 2024 - 26.5K 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/compiler/mlir/quantization/common/uniform_quantized_types_test.cc
} TEST_F(CreateI8F32UniformQuantizedTypeTest, F32ExpressedTypeSucceeds) { const UniformQuantizedType quantized_type = CreateI8F32UniformQuantizedType(UnknownLoc::get(&ctx_), ctx_, /*scale=*/1.0, /*zero_point=*/0); EXPECT_TRUE(quantized_type.getExpressedType().isF32()); } TEST_F(CreateI8F32UniformQuantizedTypeTest, SignedQuantizedTypeSucceeds) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints_test.cc
EXPECT_TRUE(failed(TryCast<DotGeneralOp>(nullptr, /*name=*/"nullptr"))); } TEST_F(AttrsAndConstraintsTest, I64ValueInI32RangeAreCastedCorrectly) { EXPECT_TRUE(succeeded(CastI64ToI32(llvm::minIntN(32)))); EXPECT_TRUE(succeeded(CastI64ToI32(llvm::maxIntN(32)))); } TEST_F(AttrsAndConstraintsTest, CastingFailsForI64ValueOutOfI32Range) { EXPECT_TRUE(failed(CastI64ToI32(llvm::minIntN(32) - 10)));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 22.9K 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/quantization/stablehlo/ops/stablehlo_op_quant_spec_test.cc
return %0 : tensor<1x3xf32> } } )mlir"; TEST_F(IsOpQuantizableStableHloTest, ConstantOpQuantizable) { OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kModuleConstantAdd); ASSERT_TRUE(module_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 04 07:19:09 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_platform_info_test.cc
} DeviceSetup device_setup_; }; class StubDevice : public DeviceBase { public: StubDevice() : DeviceBase(nullptr) {} }; #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM TEST_F(XlaPlatformInfoTest, BuildXlaDeviceCompilerXlaDeviceMetadata) { device_setup_.AddDevicesAndSetUp({DEVICE_XLA_GPU}); Device* device = device_setup_.GetDevice(DEVICE_XLA_GPU); const XlaDevice::Metadata* metadata = nullptr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Jan 14 15:17:12 UTC 2024 - 13.6K bytes - Viewed (0) -
tensorflow/c/while_loop_test.cc
// Run the graph Run({0, 0}); ExpectOutputValue(0, 8); ExpectOutputValue(1, 3); } TEST_F(CApiWhileLoopTest, UnsetCondOutput) { Init(1); params_->body_outputs[0] = params_->body_inputs[0]; ExpectError(TF_INVALID_ARGUMENT, "TF_WhileParams `cond_output` field isn't set"); } TEST_F(CApiWhileLoopTest, WrongCondOutputType) { Init(1); params_->cond_output = params_->cond_inputs[0];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 11 06:05:56 UTC 2024 - 15.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call_test.cc
%0 = stablehlo.dot_general %arg0, %arg1, contracting_dims = [1] x [0], precision = [DEFAULT, DEFAULT] : (tensor<1x1024xf32>, tensor<1024x3xf32>) -> tensor<1x3xf32> return %0 : tensor<1x3xf32> } } )mlir"; TEST_F(LiftAsFunctionCallTest, LiftedFunctionSucceeds) { OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kModuleLifted); ASSERT_TRUE(module_op); auto composite_dot_general_fn =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 26.2K bytes - Viewed (0)