- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for cases (0.15 sec)
-
tensorflow/c/c_api_experimental.cc
// threadpool of GPU event mgr, as that can trigger more callbacks to be // scheduled on that same threadpool, causing a deadlock in cases where the // caller of event_mgr->ThenExecute() blocks on the completion of the callback // (as in the case of ConstOp kernel creation on GPU, which involves copying a // CPU tensor to GPU). // Setting a larger thread pool does not help with the Swift caller, as we use
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/c_api.cc
SINGLE_CASE(kF, TF_ATTR_FLOAT, -1); SINGLE_CASE(kB, TF_ATTR_BOOL, -1); SINGLE_CASE(kType, TF_ATTR_TYPE, -1); SINGLE_CASE(kShape, TF_ATTR_SHAPE, attr->shape().unknown_rank() ? -1 : attr->shape().dim_size()); SINGLE_CASE(kTensor, TF_ATTR_TENSOR, -1); #undef SINGLE_CASE case tensorflow::AttrValue::kList: metadata.is_list = 1; metadata.list_size = 0;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
} private: bool first_call_ = true; }; // After the graph pass is registered, it takes effect globally and can affect // other test cases. Define a static variable to switch it on and off. bool GraphErrorInjectionPass::enabled_ = false; // Test to ensure that a registered graph optimization pass is only executed
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.cc
switch (data_type) { case TF_DataType::TF_BOOL: dtype.code = DLDataTypeCode::kDLBool; break; case TF_DataType::TF_HALF: case TF_DataType::TF_FLOAT: case TF_DataType::TF_DOUBLE: dtype.code = DLDataTypeCode::kDLFloat; break; case TF_DataType::TF_INT8: case TF_DataType::TF_INT16: case TF_DataType::TF_INT32: case TF_DataType::TF_INT64:
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
if (TF_GetCode(status) != TF_OK) return; // unwrap op_ and set step_id only if valid step id value was set. // Currently only required for non-TFRT use cases, e.g., EagerOp. if (step_id_.has_value()) { tensorflow::unwrap(op_.get())->SetStepId(step_id_.value()); } TFE_Execute(op_.get(), unwrapped_results.data(), &real_num_outputs, status);
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/c_api_test.cc
ScalarFloatFromTensor(fetchValues[1], &dxy_dyValue); EXPECT_EQ(X, dxy_dyValue); TF_DeleteTensor(fetchValues[0]); TF_DeleteTensor(fetchValues[1]); } // REGISTER_OP for CApiAttributesTest test cases. // Registers two ops, each with a single attribute called 'v'. // The attribute in one op will have a type 'type', the other // will have list(type). #define ATTR_TEST_REGISTER_OP(type) \
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3)