- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for combine (0.14 sec)
-
tensorflow/c/experimental/gradients/math_grad_test.cc
INSTANTIATE_TEST_SUITE_P( UnifiedCAPI, CppGradients, ::testing::Combine(::testing::Values("graphdef", "mlir"), /*tfrt*/ ::testing::Values(false), /*use_function*/ ::testing::Values(true, false))); #else INSTANTIATE_TEST_SUITE_P( UnifiedCAPI, CppGradients, ::testing::Combine(::testing::Values("graphdef", "mlir"), /*tfrt*/ ::testing::Values(false),
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Thu Apr 13 17:32:14 GMT 2023 - 16.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
// Try to combine these values into a single parallel tensor. std::array<TFE_TensorHandle*, 2> components{size_two.get(), size_three.get()}; TensorHandlePtr combined_value = CreatePerDeviceValues( context.get(), components, device_name, status.get()); ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); int num_axes = TFE_TensorHandleNumDims(combined_value.get(), status.get());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 29.3K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
TF_Status status; PartialTensorShape combined_shape; TF_RETURN_IF_ERROR(unwrap(tensors_[0].get())->Shape(&combined_shape)); for (const TensorHandlePtr& component : tensors_) { PartialTensorShape component_shape; TF_RETURN_IF_ERROR(unwrap(component.get())->Shape(&component_shape)); if (combined_shape.dims() < 0 || combined_shape.dims() != component_shape.dims()) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
std::array<TFE_TensorHandle*, 2> components{value_one.get(), value_two.get()}; TensorHandlePtr combined_value = CreatePerDeviceValues(context, components, device_name, status.get()); variable->AssignAdd(context, combined_value.get(), status.get()); } // Read the variable and verify that each component has the right modified // value. {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_test.cc
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri May 19 21:44:52 GMT 2023 - 39.1K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
/*device_mgr_owned*/ true, std::move(r), /*cluster_flr=*/nullptr, /*collective_executor_mgr=*/nullptr, /*run_eager_op_as_function=*/opts->run_eager_op_as_function, /*jit_compile_rewrite=*/opts->jit_compile_rewrite); #if !defined(IS_MOBILE_PLATFORM) eager_context->SetDistributedManager( std::make_unique<tensorflow::EagerContextDistributedManager>( eager_context));
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2)