Searched defs:group_params (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | collective_param_resolver_distributed.cc | 143 const DeviceAttributes& device, CollGroupParams* group_params, in CompleteGroupAsync()
274 const DeviceAttributes& device, CollGroupParams* group_params, in CompleteGroupDistributed()
|
| H A D | worker.cc | 410 auto group_params = new CollGroupParams(); in CompleteGroupAsync() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | collective_param_resolver_local.cc | 58 const DeviceAttributes& device, CollGroupParams* group_params, in CompleteGroupAsync()
131 const DeviceAttributes& device, CollGroupParams* group_params, in CompleteGroupLocal()
|
| H A D | test_collective_executor_mgr.h | 50 CollGroupParams* group_params, in CompleteGroupAsync()
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/mutators/ |
| H A D | GroupedConvolutionMutator.cpp | 71 NodeParams group_params = params; in create_grouped_convolution() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | collective.h | 333 CollGroupParams* group_params, in CompleteGroupAsync()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | collective_ops.cc | 1003 auto group_params = new CollGroupParams(); in ComputeAsync() local
|