- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 23 for note (0.19 sec)
-
tensorflow/c/c_api.cc
// added since the last TF_SessionRun() call. for (auto id = session->last_num_graph_nodes; id < num_nodes; ++id) { Node* const node = graph.FindNodeId(id); if (node != nullptr && node->IsOp()) { NodeDef* const node_def = graph_def.add_node(); *node_def = node->def(); } } *graph_def.mutable_library() = graph.flib_def().ToProto();
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K 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 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/not_differentiable.cc
/* Copyright 2020 The TensorFlow Authors. All Rights Reserved. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS,
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Jun 15 01:15:58 GMT 2022 - 1.3K bytes - Viewed (0) -
tensorflow/c/c_test_util.cc
} } return found_dtype && found_value; } bool IsAddN(const tensorflow::NodeDef& node_def, int n) { if (node_def.op() != "AddN" || node_def.name() != "add" || node_def.input_size() != n) { return false; } bool found_t = false; bool found_n = false; for (const auto& attr : node_def.attr()) { if (attr.first == "T") { if (attr.second.type() == tensorflow::DT_INT32) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (2) -
tensorflow/c/env.cc
options->guard_size = 0; options->numa_node = -1; } TF_Thread* TF_StartThread(const TF_ThreadOptions* options, const char* thread_name, void (*work_func)(void*), void* param) { ::tensorflow::ThreadOptions cc_options; cc_options.stack_size = options->stack_size; cc_options.guard_size = options->guard_size; cc_options.numa_node = options->numa_node;
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Aug 11 01:20:50 GMT 2021 - 7K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
#include "tensorflow/c/eager/tfe_tensorhandle_internal.h" #include "tensorflow/c/tf_status_internal.h" #include "tensorflow/core/lib/core/status_test_util.h" #include "tensorflow/core/platform/test.h" // NOTE(allenl): These tests currently go through TFE_Execute and so are // integration testing rather than purely testing the parallel device. They // correspond fairly well to the implementation, but testing the C++ directly is
C++ - Registered: Tue Apr 23 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_testlib.cc
#include "tensorflow/c/c_api.h" #include "tensorflow/c/c_api_experimental.h" #include "tensorflow/c/eager/c_api.h" #include "tensorflow/c/eager/c_api_experimental.h" #include "tensorflow/core/platform/test.h" // NOTE(allenl): These tests currently go through TFE_Execute and so are // integration testing rather than purely testing the parallel device. They // correspond fairly well to the implementation, but testing the C++ directly is
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
EXPECT_NE(TF_GetCode(status.get()), TF_CANCELLED); EXPECT_EQ(TF_GetCode(status.get()), TF_INVALID_ARGUMENT); EXPECT_THAT(TF_Message(status.get()), HasSubstr("assertion failed")); // Note that future collectives with the same context do not work at the // moment; once canceled, the collective executor requires the program to be // restarted / context to be reset. } TEST(PARALLEL_DEVICE_LIB, TestDifferentShapes) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
using tensorflow::FunctionDef; using tensorflow::Node; using tensorflow::NodeBuilder; using tensorflow::Status; using tensorflow::errors::InvalidArgument; namespace { typedef std::unique_ptr<TF_Function, decltype(&TF_DeleteFunction)> UniqueFuncPtr; } // struct TF_Operation { tensorflow::Node node; }; static TF_Operation* ToTF_Operation(Node* node) { return static_cast<TF_Operation*>(static_cast<void*>(node));
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/c_api_function.cc
TF_EXCLUSIVE_LOCKS_REQUIRED(fn_body->mu) { if (num_opers == -1) { for (const Node* node : fn_body->graph.op_nodes()) { const auto& iter = input_nodes.find(node); if (iter == input_nodes.end()) { // This node is not referenced in inputs. Add it to the body. body_nodes->push_back(node); } else { // This node is referenced in inputs. Currently, we place an
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 13.6K bytes - Viewed (2)