- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for notes (0.16 sec)
-
tensorflow/c/c_api.cc
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/c_api_experimental.cc
// A vector to keep track of newly created `tf::Tensor` objects. std::vector<Tensor> all_input_tensors; // Update the vector with information from `input_tensors` if provided. if (input_tensors != nullptr) { // Note that we take the address of the elements in `all_input_tensors` // below. Allocate enough space so that no reallocation happens, which will // make the pointers invalid. all_input_tensors.reserve(num_inputs);
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/eager/dlpack.cc
bool valid = true; int64_t expected_stride = 1; for (int i = ndim - 1; i >= 0; --i) { // Empty tensors are always compact regardless of strides. if (shape_arr[i] == 0) return true; // Note that dimensions with size=1 can have any stride. if (shape_arr[i] != 1 && stride_arr[i] != expected_stride) { valid = false; } expected_stride *= shape_arr[i]; } return valid; }
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/c_api_function.cc
} } // Compute body nodes. std::vector<const Node*> body_nodes; status->status = tensorflow::ComputeBodyNodes( fn_body, fn_name, num_opers, opers, input_nodes, &body_nodes); if (TF_GetCode(status) != TF_OK) return nullptr; // Compute body nodes. std::vector<const Node*> control_output_nodes; control_output_nodes.reserve(ncontrol_outputs);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 13.6K bytes - Viewed (2) -
tensorflow/c/eager/c_api_distributed_test.cc
// Inject failure to function instantiation if finding a node that contains // the given node name (error_node_) and requested device (error_device_). for (const auto node : graph->get()->nodes()) { if (node->name().find(error_node_) != string::npos && node->requested_device() == error_device_) { return tensorflow::errors::Internal("Injected graph pass error."); } }
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/c_api_experimental.cc
status->status = absl::OkStatus(); } TFE_Executor* TFE_NewExecutor(bool is_async, bool enable_streaming_enqueue, int in_flight_nodes_limit) { return new TFE_Executor(is_async, enable_streaming_enqueue, in_flight_nodes_limit); } void TFE_DeleteExecutor(TFE_Executor* executor) { delete executor; } bool TFE_ExecutorIsAsync(TFE_Executor* executor) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 35.9K bytes - Viewed (3) -
tensorflow/c/eager/c_api.cc
TFE_Context* ctx, TFE_ContextDevicePlacementPolicy policy) { tensorflow::unwrap(ctx)->SetThreadLocalDevicePlacementPolicy( static_cast<tensorflow::ContextDevicePlacementPolicy>(policy)); } // Note: this function looks up a thread local policy. So it should be called in // the appropriate client thread. In particular, in async mode, it may not be // safe to call this function from the async EagerExecutor threads.
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) -
tensorflow/c/c_api_test.cc
TF_DeleteBuffer(graph_def); // Can add nodes to the imported graph without trouble. Add(feed, scalar, graph, s); ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); TF_DeleteGraph(graph); TF_DeleteStatus(s); } TEST(CAPI, ImportGraphDef_WithReturnOutputs) { TF_Status* s = TF_NewStatus(); TF_Graph* graph = TF_NewGraph(); // Create a graph with two nodes: x and 3 Placeholder(graph, s);
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)