Home
last modified time | relevance | path

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

12345

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
H A 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 …]
/aosp_15_r20/external/pytorch/test/quantization/jit/
H A Dtest_ondevice_quantization.py192 def _validate_quant_dequant_nodes(self, model, num_nodes, per_channel=0): argument
202 def _validate_calculate_qparams(self, model, num_nodes): argument
218 def _check_quant_dequant_and_calc_qparams(self, model, num_nodes): argument
259 def _validate_packed_params(self, model, num_nodes, per_channel=0): argument
289 def _validate_setattr_fp_weights(self, model, num_nodes): argument
316 def _validate_quantized_forward(self, model, num_nodes): argument
342 def _check_quantize_forward(self, model, num_nodes): argument
355 def _check_quantized_forward(self, model, num_nodes): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/lib/strings/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/nccl/
H A 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 …]
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A 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
H A Dimplementation_selector.cc344 const int num_nodes = graph_view.NumNodes(); in SelectDeviceIndex() local
388 const int num_nodes = graph_view.NumNodes(); in SelectImplementation() local
H A Ddependency_optimizer.cc536 const int num_nodes = optimized_graph_->node_size(); in TransitiveReduction() local
651 const int num_nodes = optimized_graph_->node_size(); in GroupCrossDeviceControlEdges() local
/aosp_15_r20/external/ltp/testcases/kernel/lib/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/platform/
H A Dnuma_test.cc32 int num_nodes = port::NUMANumNodes(); in TEST() local
51 int num_nodes = port::NUMANumNodes(); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dlstm.py38 def LSTMCellWeightsShape(num_inputs, num_nodes): argument
135 def BuildLSTMLayer(batch_size, seq_length, num_inputs, num_nodes): argument
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dimpl.cpp405 const auto num_nodes = static_cast<uint32_t>(nodes.size()); in ManagedTensorRanges() local
1473 const auto num_nodes = static_cast<uint32_t>(nodes_.size()); in benchmark() local
1698 const auto num_nodes = static_cast<uint32_t>(nodes_.size()); in benchmark_individual_ops() local
1774 const auto num_nodes = static_cast<uint32_t>(nodes_.size()); in benchmark_individual_ops() local
1814 const auto num_nodes = static_cast<uint32_t>(nodes_.size()); in benchmark_individual_ops() local
1853 const auto num_nodes = static_cast<uint32_t>(nodes_.size()); in check_for_memory_leak() local
1933 const auto num_nodes = static_cast<uint32_t>(nodes_.size()); in checkOutputTensorMemoryLeaks() local
/aosp_15_r20/external/mesa3d/src/intel/vulkan/grl/gpu/
H A Dcommon.h126 GRL_INLINE uint allocate_inner_nodes(global struct BVHBase* base, uint num_nodes ) in allocate_inner_nodes()
130 GRL_INLINE uint allocate_procedural_leaves(global struct BVHBase* base, uint num_nodes) in allocate_procedural_leaves()
135 GRL_INLINE uint allocate_quad_leaves(global struct BVHBase* base, uint num_nodes) in allocate_quad_leaves()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpytree.h164 int num_nodes() const { return traversal_.size(); } in num_nodes() function
200 int num_nodes = 0; member
/aosp_15_r20/external/mesa3d/src/util/tests/
H A Ddag_test.cpp79 init_nodes(struct dag *dag, struct node *nodes, unsigned num_nodes) in init_nodes()
87 #define INIT_NODES(num_nodes) \ argument
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/internal/
H A Dtfprof_stats.cc202 size_t num_nodes = nodes_map_.size(); in AddGraph() local
297 size_t num_nodes = nodes_map_.size(); in AddRunMeta() local
/aosp_15_r20/external/arm-trusted-firmware/plat/xilinx/zynqmp/pm_service/
H A Dpm_api_clock.c230 uint8_t num_nodes; member
2481 uint8_t num_nodes; in pm_api_clock_get_topology() local
2533 uint8_t num_nodes; in pm_api_clock_get_fixedfactor_params() local
/aosp_15_r20/external/trusty/arm-trusted-firmware/plat/xilinx/zynqmp/pm_service/
Dpm_api_clock.c234 uint8_t num_nodes; member
2490 uint8_t num_nodes; in pm_api_clock_get_topology() local
2547 uint8_t num_nodes; in pm_api_clock_get_fixedfactor_params() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dgraph_view.cc86 const int num_nodes = graph->node_size(); in GraphView() local
466 const int num_nodes = graph->node_size(); in MutableGraphView() local
504 const int num_nodes = nodes_.size(); in CheckFaninsInternal() local
542 const int num_nodes = nodes_.size(); in AddFaninsInternal() local
1436 const int num_nodes = nodes_.size(); in SortTopologically() local
/aosp_15_r20/external/libchrome/base/containers/
H A Dlinked_list_unittest.cc50 int num_nodes, const int* node_ids, bool forward) { in ExpectListContentsForDirection()
64 int num_nodes, in ExpectListContents()
/aosp_15_r20/external/cronet/base/containers/
H A Dlinked_list_unittest.cc51 int num_nodes, const int* node_ids, bool forward) { in ExpectListContentsForDirection()
65 int num_nodes, in ExpectListContents()
/aosp_15_r20/external/mesa3d/src/imgui/
H A Dimstb_rectpack.h186 int num_nodes; member
260 … stbrp_init_target(stbrp_context *context, int width, int height, stbrp_node *nodes, int num_nodes) in stbrp_init_target()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/
H A Dgraph_view_test.cc296 const int num_nodes = state.range(0); in BM_GraphViewConstruction() local
338 const int num_nodes = state.range(0); in BM_GraphViewGetNode() local
/aosp_15_r20/external/federated-compute/fcp/secagg/server/
H A Dsecret_sharing_graph_factory.h40 int num_nodes, int threshold) { in CreateCompleteGraph()

12345