Searched defs:index_copy (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_xla_computations_pass_test.cc | 287 std::unordered_map<string, Node*> index_copy = in TEST() local
|
| /external/pytorch/aten/src/ATen/native/ |
| D | NamedTensor.cpp | 356 Tensor index_copy(const Tensor& self, Dimname dim, const Tensor& index, const Tensor& source) { in index_copy() function
|
| D | TensorAdvancedIndexing.cpp | 263 TORCH_PRECOMPUTE_META_FUNC(index_copy) in TORCH_PRECOMPUTE_META_FUNC() argument
|
| /external/pytorch/test/ |
| D | test_decomp.py | 599 def index_copy(xs, x): function
|
| /external/pytorch/torch/onnx/ |
| D | symbolic_opset11.py | 959 def index_copy(g: jit_utils.GraphContext, self, dim, index, source): function
|
| D | symbolic_opset9.py | 2970 def index_copy(g: jit_utils.GraphContext, self, dim, index, source): function
|
| /external/pytorch/torch/_refs/ |
| D | __init__.py | 3961 def index_copy(x: TensorLike, dim: int, index: TensorLike, tensor: TensorLike): function
|
| /external/pytorch/torch/_decomp/ |
| D | decompositions.py | 2667 def index_copy(x: TensorLike, dim: int, index: TensorLike, tensor: TensorLike): function
|