| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 20 group_size, argument
96 group_size, argument
160 group_size, argument
200 group_size, argument
251 group_size, argument
313 group_size, argument
348 group_size, argument
392 group_size, argument
429 group_size, argument
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | group_assignment.h | 117 int group_size() const { return global_.group_size(); } in group_size() function
130 StatusOr<int> group_size(int slice_id) const { in group_size() function
157 int group_size() const { return replica_ids_.front().size(); } in group_size() function
|
| D | dtensor_mixed_precision_reduce.cc | 41 int32* group_size) { in GetAllReduceGroupSize()
77 int32 group_size; in MaybeUpcastForReduction() local
|
| D | group_assignment_test.cc | 42 int group_size = replica_ids.front().size(); in CreateGroupAssignmentAttr() local
|
| /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 | 47 def all_reduce(t, group_size, group_key, instance_key, *args, **kwargs): argument
53 def all_gather(t, group_size, group_key, instance_key, *args, **kwargs): argument
65 def all_reduce(t, group_size, group_key, instance_key, *args, **kwargs): argument
73 def all_gather(t, group_size, group_key, instance_key, *args, **kwargs): argument
81 def broadcast_send(t, shape, dtype, group_size, group_key, instance_key, argument
90 def broadcast_recv(shape, dtype, group_size, group_key, instance_key, *args, argument
452 def run_all_reduce(group_size, group_key): argument
470 def run_and_assert(group_size, group_key): argument
1259 def run(group_size, reported_group_size=None): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_ops.cc | 509 Tensor* group_size = nullptr; in Compute() local
527 const int32_t base_key, Tensor* group_size, in ComputeGroupKey()
590 const Tensor& group_size, const Tensor& group_key, in FillCollectiveParams()
910 CollectiveGroupResource(int32 group_key, int32 rank, int32 group_size, in CollectiveGroupResource()
932 int32 group_size() const { return group_size_; } in group_size() function in tensorflow::__anon8b310e240111::CollectiveGroupResource
967 auto group_size = group_size_t.unaligned_flat<int32>()(0); in CheckInputs() local
982 auto group_size = group_size_t.unaligned_flat<int32>()(0); in ComputeAsync() local
1070 int64 group_size; in FillCollectiveParams() local
|
| D | collective_nccl_reducer.cc | 27 Tensor group_size; in Run() local
|
| D | sparse_softmax_op.cc | 102 const int group_size = group_vals.size(); in Compute() local
|
| D | bias_op_gpu.cu.cc | 156 int32 batch, int32 bias_size, int32 image_size, int group_size) { in BiasGradNCHW_SharedAtomics()
242 int group_size = (config.block_count + bias_size - 1) / bias_size; in compute() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_collective.cc | 77 LogicalResult SetCollectiveInfo(IntegerAttr group_size, IntegerAttr group_key, in SetCollectiveInfo()
100 IntegerAttr group_size = builder.getI32IntegerAttr(replica_groups.size()); in SetCollectiveInfo() local
302 const auto group_size = group_size_attr.getSplatValue<IntegerAttr>(); in matchAndRewrite() local
|
| /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/tensorflow/tensorflow/compiler/xla/service/ |
| D | reduce_scatter_utils.h | 29 int64_t group_size; member
|
| D | all_reduce_simplifier.cc | 82 int64_t group_size = get_replica_group_size(inst); in Run() local
|
| D | convolution_group_converter.cc | 118 std::vector<int32_t> GetMaskIds(int64_t group_size, int64_t group_count) { in GetMaskIds()
176 int64_t group_size = filter_shape.dimensions(kernel_input_feature_dim); in GetExpandedFilterMask() local
441 int64_t group_size = filter->shape().dimensions(kernel_input_feature_dim); in HandleConvolution() local
|
| D | reduce_scatter_utils.cc | 130 const MapIdToTableOffset& map_id, int64_t group_size, in IsPerIdOffset()
321 int64_t group_size; in MatchReduceScatter() 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/tests/ |
| D | grouped_convolution_test.cc | 41 int64_t group_size, group_count; member
72 int64_t group_size = option[5]; in GetConv2DTestCases() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | collective_ops.cc | 69 int group_size; in __anone3133c620102() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | all_reduce_op.cc | 39 int64_t group_key, group_size; in Compile() local
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 548 int32 group_size = 2; field
559 int32 group_size = 2; field
578 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/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.h | 166 int group_size; member
|