| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 24 def all_reduce(t, group_size, group_key, instance_key, merge_op, final_op, argument
61 def all_gather(t, group_size, group_key, instance_key): argument
90 def broadcast_send(t, shape, dtype, group_size, group_key, instance_key): argument
137 def broadcast_recv(shape, dtype, group_size, group_key, instance_key): argument
|
| /external/mesa3d/src/mesa/state_tracker/ |
| D | st_cb_compute.c | 39 const GLuint *group_size, in st_dispatch_compute_common()
89 const GLuint *group_size) in st_dispatch_compute_group_size()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_batch_matmul_op.cc | 115 std::vector<MKL_INT> group_size(1, batch_size); in Compute() local
144 const MKL_INT group_count, const MKL_INT *group_size) { in MklCblasGemmBatch()
163 const MKL_INT group_count, const MKL_INT *group_size) { in MklCblasGemmBatch()
182 const MKL_INT group_count, const MKL_INT *group_size) { in MklCblasGemmBatch()
204 const MKL_INT group_count, const MKL_INT *group_size) { in MklCblasGemmBatch()
|
| D | sparse_softmax_op.cc | 103 const int group_size = group_vals.size(); in Compute() local
|
| D | collective_nccl_reducer.cc | 82 Tensor group_size; in Run() local
|
| D | bias_op_gpu.cu.cc | 151 int group_size) { in BiasGradNCHW_SharedAtomics()
224 int group_size = (config.block_count + bias_size - 1) / bias_size; in compute() local
|
| D | roll_op.cc | 160 int64 group_size = 0; in DoRollWithMemcpy() local
|
| D | random_op.cc | 687 const size_t group_size = device.maxSyclThreadsPerBlock(); in operator ()() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | grouped_convolution_test.cc | 37 int64 group_size, group_count; member
67 int64 group_size = option[5]; in GetConv2DTestCases() local
|
| /external/mesa3d/src/mesa/main/ |
| D | compute.c | 104 const GLuint *group_size) in validate_DispatchComputeGroupSizeARB()
319 const GLuint group_size[3] = { group_size_x, group_size_y, group_size_z }; in dispatch_compute_group_size() local
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 500 int32 group_size = 2; field
508 int32 group_size = 2; field
525 int32 group_size = 6; field
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 116 std::vector<int32> GetMaskIds(int64 group_size, int64 group_count) { in GetMaskIds()
173 int64 group_size = filter_shape.dimensions(kernel_input_feature_dim); in GetExpandedFilterMask() local
337 int64 group_size = filter->shape().dimensions(kernel_input_feature_dim); in HandleConvolution() local
|
| /external/mesa3d/src/mesa/drivers/dri/i965/ |
| D | brw_compute.c | 124 unsigned group_size = prog_data->local_size[0] * in brw_emit_gpgpu_walker() local
|
| /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/python/distribute/ |
| D | cross_device_utils.py | 169 def group_device_names(devices, 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
|
| D | ring_gatherer_test.cc | 396 int group_size = col_params_.group.group_size; in DeviceInstance() local
|
| D | ring_reducer_test.cc | 420 int group_size = col_params_.group.group_size; in DeviceInstance() local
|
| D | hierarchical_tree_broadcaster_test.cc | 577 int group_size = col_params_.group.group_size; in DeviceInstance() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 54 int32 group_size; member
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | group_by_window_dataset_op.cc | 423 int64 group_size; in RestoreGroup() local
|
| /external/mesa3d/src/intel/common/ |
| D | gen_decoder.h | 104 uint32_t group_size; member
|
| /external/mesa3d/src/intel/vulkan/ |
| D | genX_pipeline.c | 1797 uint32_t group_size = cs_prog_data->local_size[0] * local
|
| /external/mesa3d/src/gallium/drivers/r600/ |
| D | evergreen_compute.c | 582 int group_size = 1; in evergreen_emit_dispatch() local
|