Home
last modified time | relevance | path

Searched defs:num_nodes (Results 1 – 25 of 102) sorted by relevance

12345

/external/tensorflow/tensorflow/core/lib/strings/
Dproto_serialization_test.cc31 GraphDef MakeGraphDef(int num_nodes) { in MakeGraphDef()
45 int num_nodes = state.range(0); in BM_ProtoSerializationToString() local
59 int num_nodes = state.range(0); in BM_ProtoSerializationToBuffer() local
73 int num_nodes = state.range(0); in BM_DeterministicProtoHash64() local
84 int num_nodes = state.range(0); in BM_AreSerializedProtosEqual() local
/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
Dclient_server_test.cc65 void StartService(int num_nodes, bool use_coordination_service, in StartService()
117 int num_nodes = 3; in TEST_P() local
264 int num_nodes = 2; in TEST_P() local
299 int num_nodes = 3; in TEST_P() local
349 int num_nodes = 3; in TEST_P() local
386 int num_nodes = 3; in TEST_P() local
443 int num_nodes = 3; in TEST_P() local
477 int num_nodes = 3; in TEST_P() local
511 int num_nodes = 2; in TEST_P() local
541 int num_nodes = 2; in TEST_P() local
[all …]
/external/tensorflow/tensorflow/core/platform/
Dnuma_test.cc32 int num_nodes = port::NUMANumNodes(); in TEST() local
51 int num_nodes = port::NUMANumNodes(); in TEST() local
/external/ltp/testcases/kernel/lib/
Dnuma_helper.c152 int get_allowed_nodes_arr(int flag, int *num_nodes, int **nodes) in get_allowed_nodes_arr()
220 int num_nodes = 0; in get_allowed_nodes() local
247 int i, ret, num_nodes; in print_node_info() local
/external/tensorflow/tensorflow/compiler/tests/
Dlstm.py38 def LSTMCellWeightsShape(num_inputs, num_nodes): argument
135 def BuildLSTMLayer(batch_size, seq_length, num_inputs, num_nodes): argument
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer.cc129 const int num_nodes = context->num_nodes; in ExpandLayoutSensitiveOp() local
151 const int num_nodes = context->num_nodes; in ExpandLayoutAgnosticOp() local
238 const int num_nodes = graph_view->NumNodes(); in EraseCancellableNodes() local
287 const int num_nodes = graph_view->NumNodes(); in EraseCancellableNodesAroundPad() local
392 const int num_nodes = graph_view->NumNodes(); in EraseOutputShapeAttrs() local
Dimplementation_selector.cc344 const int num_nodes = graph_view.NumNodes(); in SelectDeviceIndex() local
388 const int num_nodes = graph_view.NumNodes(); in SelectImplementation() local
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager_test.cc59 const int num_nodes; member
88 static void PopulateMultiNodeParams(int* num_nodes, int* num_ranks_per_node) { in PopulateMultiNodeParams()
104 TestCase* MakeReductionTestCase(int num_nodes, int num_ranks_per_node, in MakeReductionTestCase()
165 TestCase* MakeGatherTestCase(int num_nodes, int num_ranks_per_node, in MakeGatherTestCase()
210 TestCase* MakeBroadcastTestCase(int num_nodes, int num_ranks_per_node, in MakeBroadcastTestCase()
302 void RunMultiNodeAllReduceTest(const int num_nodes, in RunMultiNodeAllReduceTest()
310 const int num_nodes = node_states.size(); in RunMultiNodeAllReduceTest() local
362 void RunMultiNodeBroadcastTest(const int num_nodes, in RunMultiNodeBroadcastTest()
650 int num_nodes; in TYPED_TEST() local
673 int num_nodes; in TYPED_TEST() local
[all …]
/external/libtextclassifier/native/utils/sentencepiece/
Dtest_utils.cc35 const int num_nodes = trie_blob_size / sizeof(TrieNode); in NormalizerFromSpec() local
/external/webrtc/modules/audio_processing/transient/
Dwpd_tree.h78 int num_nodes() const { return num_nodes_; } in num_nodes() function
/external/tensorflow/tensorflow/compiler/xla/python/
Dpytree.h164 int num_nodes() const { return traversal_.size(); } in num_nodes() function
200 int num_nodes = 0; member
/external/angle/third_party/abseil-cpp/absl/synchronization/internal/
Dgraphcycles_benchmark.cc27 const int num_nodes = state.range(0); in BM_StressTest() local
/external/libtextclassifier/abseil-cpp/absl/synchronization/internal/
Dgraphcycles_benchmark.cc27 const int num_nodes = state.range(0); in BM_StressTest() local
/external/abseil-cpp/absl/synchronization/internal/
Dgraphcycles_benchmark.cc27 const int num_nodes = state.range(0); in BM_StressTest() local
/external/cronet/third_party/abseil-cpp/absl/synchronization/internal/
Dgraphcycles_benchmark.cc27 const int num_nodes = state.range(0); in BM_StressTest() local
/external/webrtc/third_party/abseil-cpp/absl/synchronization/internal/
Dgraphcycles_benchmark.cc27 const int num_nodes = state.range(0); in BM_StressTest() local
/external/rust/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/synchronization/internal/
Dgraphcycles_benchmark.cc27 const int num_nodes = state.range(0); in BM_StressTest() local
/external/openscreen/third_party/abseil/src/absl/synchronization/internal/
Dgraphcycles_benchmark.cc27 const int num_nodes = state.range(0); in BM_StressTest() local
/external/tensorflow/third_party/absl/abseil-cpp/absl/synchronization/internal/
Dgraphcycles_benchmark.cc27 const int num_nodes = state.range(0); in BM_StressTest() local
/external/icing/proto/icing/legacy/index/proto/
Dicing-dynamic-trie-header.proto32 optional uint32 num_nodes = 6; field
/external/tensorflow/tensorflow/core/grappler/
Dgraph_topology_view.h67 int num_nodes() const { return num_nodes_; } in num_nodes() function
/external/tensorflow/tensorflow/core/graph/
Dgraph_test.cc665 const int num_nodes = 10; in TEST_F() local
678 const int num_nodes = state.range(0); in BM_InEdgeIteration() local
718 const int num_nodes = state.range(0); in BM_GraphCreation() local
757 const int num_nodes = state.range(0); in BM_ToGraphDef() local
796 const int num_nodes = state.range(0); in BM_RemoveNode() local
/external/marisa-trie/lib/marisa/grimoire/trie/
Dlouds-trie.h38 std::size_t num_nodes() const { in num_nodes() function
/external/tensorflow/tensorflow/compiler/xla/
Dshape_tree_test.cc65 int num_nodes = 0; in TestShapeConstructor() local
91 int num_nodes = 0; in TestInitValueConstructor() local
390 int num_nodes = 0; in TEST_F() local
413 int num_nodes = 0; in TEST_F() local
423 int num_nodes = 0; in TEST_F() local
/external/ltp/testcases/kernel/syscalls/migrate_pages/
Dmigrate_pages03.c41 static int num_nodes, max_node; variable

12345