Searched defs:index_fill (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | NamedTensor.cpp | 344 Tensor index_fill(const Tensor& self, Dimname dim, const Tensor& index, const Scalar& source) { in index_fill() function
350 Tensor index_fill(const Tensor& self, Dimname dim, const Tensor& index, const Tensor& source) { in index_fill() function
|
| D | TensorAdvancedIndexing.cpp | 1563 Tensor index_fill(const Tensor & self, int64_t dim, const Tensor & index, const Scalar& source) { in index_fill() function
1567 Tensor index_fill(const Tensor & self, int64_t dim, const Tensor & index, const Tensor & source) { in index_fill() function
|
| /external/pytorch/torch/onnx/ |
| D | symbolic_opset11.py | 947 def index_fill(g: jit_utils.GraphContext, self, dim, index, value): function
|
| D | symbolic_opset9.py | 2957 def index_fill(g: jit_utils.GraphContext, self, dim, index, value): function
|
| /external/pytorch/torch/_refs/ |
| D | __init__.py | 3982 def index_fill( function
|