Home
last modified time | relevance | path

Searched refs:tensor_creation_op (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/csrc/jit/runtime/
Dregister_special_ops.cpp259 #define DEFINE_TORCH_TENSOR_OP(operator_type, c_type, tensor_creation_op) \ argument
271 auto tensor = tensor_creation_op; \
286 auto tensor = tensor_creation_op; \
/external/pytorch/torch/distributed/_shard/sharded_tensor/_ops/
Dinit.py128 def tensor_creation_op(types, args=(), kwargs=None, pg=None): function