| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 24 group_size, argument
75 group_size, argument
124 group_size, argument
163 group_size, argument
206 group_size, argument
265 group_size, argument
300 group_size, argument
343 group_size, argument
|
| /external/mesa3d/src/mesa/state_tracker/ |
| D | st_cb_compute.c | 40 const GLuint *group_size, in st_dispatch_compute_common()
90 const GLuint *group_size) in st_dispatch_compute_group_size()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_test.py | 56 def all_reduce(t, group_size, group_key, instance_key, *args, **kwargs): argument
64 def all_gather(t, group_size, group_key, instance_key, *args, **kwargs): argument
72 def broadcast_send(t, shape, dtype, group_size, group_key, instance_key, argument
81 def broadcast_recv(shape, dtype, group_size, group_key, instance_key, *args, argument
361 def run_all_reduce(group_size, group_key): argument
377 def run_and_assert(group_size, group_key): argument
819 def run(group_size, reported_group_size=None): argument
|
| /external/mesa3d/src/mesa/drivers/dri/i965/ |
| D | brw_cs.h | 33 unsigned group_size; member
|
| D | brw_compute.c | 133 const GLuint *group_size) in brw_dispatch_compute_group_size()
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_lower_system_values.c | 46 nir_ssa_def *group_size = nir_load_local_group_size(b); in build_global_group_size() local
360 nir_ssa_def *group_size = nir_load_local_group_size(b); in lower_compute_system_value_instr() local
|
| /external/mesa3d/src/mesa/main/ |
| D | compute.c | 104 const GLuint *group_size) in validate_DispatchComputeGroupSizeARB()
354 const GLuint group_size[3] = { group_size_x, group_size_y, group_size_z }; in dispatch_compute_group_size() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_reduce_simplifier.cc | 66 int64 group_size = get_replica_group_size(inst); in Run() local
|
| D | convolution_group_converter.cc | 115 std::vector<int32> GetMaskIds(int64 group_size, int64 group_count) { in GetMaskIds()
172 int64 group_size = filter_shape.dimensions(kernel_input_feature_dim); in GetExpandedFilterMask() local
435 int64 group_size = filter->shape().dimensions(kernel_input_feature_dim); in HandleConvolution() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | collective_ops.cc | 67 int group_size; in __anon52c2f9040102() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | grouped_convolution_test.cc | 41 int64 group_size, group_count; member
71 int64 group_size = option[5]; in GetConv2DTestCases() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer.cc | 27 Tensor group_size; in Run() local
|
| D | collective_ops.cc | 509 const Tensor& group_size = c->input(1); in ComputeAsync() local
638 const Tensor& group_size = c->input(1); in ComputeAsync() local
772 const Tensor& group_size = c->input(1); in ComputeAsync() local
897 const Tensor& group_size = c->input(0); in ComputeAsync() local
|
| D | sparse_softmax_op.cc | 99 const int group_size = group_vals.size(); in Compute() local
|
| D | bias_op_gpu.cu.cc | 153 int32 batch, int32 bias_size, int32 image_size, int group_size) { in BiasGradNCHW_SharedAtomics()
239 int group_size = (config.block_count + bias_size - 1) / bias_size; in compute() local
|
| D | roll_op.cc | 242 int64 group_size = 0; in DoRollWithMemcpy() local
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 546 int32 group_size = 2; field
557 int32 group_size = 2; field
576 int32 group_size = 6; field
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | work_group_picking.cc | 103 int GetPenalty(int grid_size, int group_size) { in GetPenalty()
108 int GetPenalty(int2 grid_size, int2 group_size) { in GetPenalty()
|
| /external/kernel-headers/original/uapi/linux/ |
| D | efs_fs_sb.h | 55 __u32 group_size; /* # of blocks a group consists of */ member
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed_test.cc | 213 int group_size = num_workers * num_devices; in IssueRequests() local
224 int group_size) { in IssueRequest()
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_test.cc | 384 int group_size, TF_Status* status) { in CollectiveSum()
470 const char* function_name, int group_size, in RegisterCollectiveMulFunction()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 265 group_size, argument
|
| /external/mesa3d/src/broadcom/cle/ |
| D | v3d_decoder.h | 72 uint32_t group_size; member
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | hierarchical_tree_broadcaster.cc | 247 int group_size = 0; in TreeSendTo() local
|
| /external/mesa3d/src/intel/compiler/ |
| D | brw_fs.h | 369 fs_reg group_size[3]; variable
|