| /external/pytorch/torch/nn/parallel/ |
| D | scatter_gather.py | 33 def scatter( function
42 def scatter( function
50 def scatter(inputs, target_gpus, dim=0): function
|
| D | comm.py | 168 def scatter(tensor, devices=None, chunk_sizes=None, dim=0, streams=None, *, out=None): function
|
| D | data_parallel.py | 201 def scatter( member in DataParallel
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_simplifier.cc | 96 HloScatterInstruction* scatter, in TransformScatterUpdates()
133 auto* scatter = Cast<HloScatterInstruction>(inst); in ExpandInstruction() local
206 if (auto* scatter = DynCast<HloScatterInstruction>(inst)) { in InstructionMatchesPattern() local
|
| D | scatter_expander.cc | 249 HloScatterInstruction* scatter, HloInstruction* induction_var, in ScatterLoopBody()
379 static int64_t ScatterTripCount(const HloScatterInstruction* scatter) { in ScatterTripCount()
414 auto* scatter = Cast<HloScatterInstruction>(inst); in ExpandInstruction() local
488 auto* scatter = DynCast<HloScatterInstruction>(inst); in InstructionMatchesPattern() local
|
| D | hlo_sharding_util.cc | 731 const HloScatterInstruction* scatter) { in ScatterIndexSharding()
774 const HloScatterInstruction* scatter) { in ScatterDataSharding()
816 const HloSharding& index_sharding, const HloScatterInstruction& scatter) { in ScatterEffectiveIndexSharding()
860 const HloScatterInstruction& scatter) { in ScatterEffectiveDataSharding()
1134 const HloSharding& update_sharding, const HloScatterInstruction& scatter) { in ScatterOutputShardingFromUpdate()
1153 const HloScatterInstruction& scatter) { in ScatterUpdateShardingFromOutput()
1173 const HloScatterInstruction& scatter) { in IdentityValueAndHloOpcodeForScatterReduceComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | gather_scatter_handler.cc | 774 const HloScatterInstruction* scatter, absl::Span<PartitionedHlo> operands, in PartitionScatterPassthroughOperand()
813 const HloScatterInstruction* scatter, absl::Span<PartitionedHlo> operands, in PartitionScatterIndexPassthroughPartition()
952 const HloScatterInstruction* scatter, absl::Span<PartitionedHlo> operands, in PartitionScatterTrivialIndexedOperandDimension()
1008 const HloScatterInstruction* scatter, absl::Span<PartitionedHlo> operands, in PartitionScatter()
1044 auto scatter = Cast<HloScatterInstruction>(hlo); in HandleScatter() local
|
| /external/pytorch/test/distributed/ |
| D | test_c10d_ops_nccl.py | 587 def scatter(output_t, input_t, rootRank): function
622 def scatter(output_t, input_t, rootRank): function
|
| /external/mesa3d/src/vulkan/runtime/radix_sort/ |
| D | radix_sort_vk.h | 102 struct rs_pipeline_layout_scatter scatter[RS_KV_DWORDS_MAX]; member
111 struct rs_pipeline_scatter scatter[RS_KV_DWORDS_MAX]; member
|
| D | target.h | 51 } scatter; member
|
| /external/intel-media-driver/media_driver/linux/ult/libdrm_mock/ |
| D | xf86drmHash_mock.c | 85 static unsigned long scatter[256]; in HashHash() local
|
| /external/intel-media-driver/media_softlet/linux/common/os/i915/ |
| D | xf86drmHash.c | 85 static unsigned long scatter[256]; in HashHash() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | tensor_array_ops.py | 334 def scatter(self, indices, value, name=None): member in _GraphTensorArray
606 def scatter(self, indices, value, name=None): member in _GraphTensorArrayV2
901 def scatter(self, indices, value, name=None): member in _EagerTensorArray
1255 def scatter(self, indices, value, name=None): member in TensorArray
|
| /external/libdrm/ |
| D | xf86drmHash.c | 85 static unsigned long scatter[256]; in HashHash() local
|
| /external/pytorch/test/cpp_extensions/ |
| D | cpp_c10d_extension.cpp | 76 c10::intrusive_ptr<Work> ProcessGroupTest::scatter( in scatter() function in c10d::ProcessGroupTest
|
| /external/pytorch/aten/src/ATen/native/ |
| D | NamedTensor.cpp | 368 Tensor scatter(const Tensor& self, Dimname dim, const Tensor& index, const Tensor& source) { in scatter() function
371 Tensor scatter(const Tensor& self, Dimname dim, const Tensor& index, const Scalar& source) { in scatter() function
|
| /external/pytorch/torch/csrc/distributed/c10d/ |
| D | FakeProcessGroup.hpp | 113 c10::intrusive_ptr<Work> scatter( in scatter() function in c10d::FakeProcessGroup
|
| D | Backend.hpp | 192 virtual c10::intrusive_ptr<Work> scatter( in scatter() function in c10d::Backend
|
| D | ProcessGroup.hpp | 325 virtual c10::intrusive_ptr<Work> scatter( in scatter() function in c10d::ProcessGroup
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tensor_array_ops_decomposition.cc | 134 } else if (auto scatter = in GetTensorArrayElementShape() local
443 TF::TensorArrayScatterV3Op scatter, in HandleTensorArrayScatterV3Op()
891 } else if (auto scatter = llvm::dyn_cast<TF::TensorArrayScatterV3Op>(&op)) { in DecomposeTensorArrayOps() local
|
| D | tensor_list_ops_decomposition.cc | 778 TF::TensorListScatterIntoExistingListOp scatter, in HandleTensorListScatterIntoExistingListOp()
858 } else if (auto scatter = in DecomposeTensorListOpsInternal() local
|
| /external/mesa3d/src/glx/ |
| D | glxhash.c | 132 static unsigned long scatter[256]; in HashHash() local
|
| /external/pytorch/torch/csrc/cuda/ |
| D | comm.cpp | 313 std::vector<at::Tensor> scatter( in _broadcast_out_impl() function
|
| /external/pytorch/test/cpp/api/ |
| D | parallel.cpp | 27 Scatter scatter( in TEST_F() local
|
| /external/pytorch/torch/distributed/nn/ |
| D | functional.py | 47 def scatter(tensors, src=0, group=group.WORLD): function
|