| /aosp_15_r20/external/pytorch/torch/csrc/autograd/functions/ |
| H A D | comm.h | 16 struct TORCH_CUDA_CU_API Scatter : public Node { struct
28 std::vector<at::Device> devices_;
29 std::optional<std::vector<int64_t>> chunk_sizes_;
30 int64_t dim_;
31 std::optional<std::vector<std::optional<at::cuda::CUDAStream>>> streams_;
32 bool unsqueeze_scalars_;
|
| H A D | comm.cpp | 16 Scatter::Scatter( in Scatter() function in torch::autograd::Scatter
|
| /aosp_15_r20/external/pytorch/torch/nn/parallel/ |
| H A D | _functions.py | 92 class Scatter(Function): class
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| H A D | operation_semantics.md | 2506 ## Scatter section in Operation Semantics
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | multi_threaded_pg.py | 156 class Scatter: class
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | list_kernels.h | 818 Status Scatter(OpKernelContext* c, const Tensor& value, const Tensor& indices, in Scatter() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_cost_analysis_test.cc | 1170 TEST_F(HloCostAnalysisTest, Scatter) { in TEST_F() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.cc | 2427 XlaOp XlaBuilder::Scatter(XlaOp input, XlaOp scatter_indices, XlaOp updates, in Scatter() function in xla::XlaBuilder
2436 XlaOp XlaBuilder::Scatter(absl::Span<const XlaOp> inputs, XlaOp scatter_indices, in Scatter() function in xla::XlaBuilder
4906 XlaOp Scatter(const XlaOp input, const XlaOp scatter_indices, in Scatter() function
4915 XlaOp Scatter(absl::Span<const XlaOp> inputs, XlaOp scatter_indices, in Scatter() function
|
| /aosp_15_r20/external/swiftshader/src/Reactor/ |
| H A D | LLVMReactor.cpp | 1221 void Scatter(RValue<Pointer<Float>> base, RValue<SIMD::Float> val, RValue<SIMD::Int> offsets, RValu… in Scatter() function
1226 void Scatter(RValue<Pointer<Int>> base, RValue<SIMD::Int> val, RValue<SIMD::Int> offsets, RValue<SI… in Scatter() function
|
| H A D | SubzeroReactor.cpp | 4163 void Scatter(RValue<Pointer<Float>> base, RValue<SIMD::Float> val, RValue<SIMD::Int> offsets, RValu… in Scatter() function
4169 void Scatter(RValue<Pointer<Int>> base, RValue<SIMD::Int> val, RValue<SIMD::Int> offsets, RValue<SI… in Scatter() function
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | ir.py | 517 class Scatter(Pointwise): class
|
| /aosp_15_r20/external/llvm/lib/CodeGen/SelectionDAG/ |
| H A D | SelectionDAGBuilder.cpp | 3809 SDValue Scatter = DAG.getMaskedScatter(DAG.getVTList(MVT::Other), VT, sdl, in visitMaskedScatter() local
|
| /aosp_15_r20/external/swiftshader/third_party/llvm-10.0/llvm/lib/CodeGen/SelectionDAG/ |
| H A D | SelectionDAGBuilder.cpp | 4487 SDValue Scatter = DAG.getMaskedScatter(DAG.getVTList(MVT::Other), VT, sdl, in visitMaskedScatter() local
|
| /aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/CodeGen/SelectionDAG/ |
| H A D | SelectionDAGBuilder.cpp | 4507 SDValue Scatter = DAG.getMaskedScatter(DAG.getVTList(MVT::Other), VT, sdl, in visitMaskedScatter() local
|
| /aosp_15_r20/external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/X86/ |
| H A D | X86ISelLowering.cpp | 44042 auto *Scatter = cast<MaskedScatterSDNode>(GorS); in combineGatherScatter() local
44071 auto *Scatter = cast<MaskedScatterSDNode>(GorS); in combineGatherScatter() local
44098 auto *Scatter = cast<MaskedScatterSDNode>(GorS); in combineGatherScatter() local
|
| /aosp_15_r20/test/mlts/benchmark/results/ |
| D | Chart.bundle.min.js | 10 …on"==typeof define&&define.amd)define([],t);else{("undefined"!=typeof window?window:"undefined"!=t…
|
| /aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/Target/X86/ |
| H A D | X86ISelLowering.cpp | 53690 auto *Scatter = cast<MaskedScatterSDNode>(GorS); in rebuildGatherScatter() local
|