Home
last modified time | relevance | path

Searched defs:to_tensor (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dcollective_rma_distributed.cc44 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvBufCall()
111 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
/aosp_15_r20/external/pytorch/torch/_export/pass_infra/
H A Dproxy_value.py30 def to_tensor(self) -> torch.Tensor: member in ProxyValue
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_conversion_ops.py44 def to_tensor(rt_input, default_value=None, name=None): function
H A Dragged_tensor.py1789 def to_tensor(self, default_value=None, name=None, shape=None): member in RaggedTensor
/aosp_15_r20/external/pytorch/torch/_numpy/
H A D_dtypes_impl.py169 def to_tensor(scalar, dtype=None): function
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcollective_rma_local.cc29 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
H A Dcollective_test_util.cc61 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_quantized.py222 def to_tensor(X, device): function
/aosp_15_r20/external/pytorch/torch/masked/maskedtensor/
H A Dcore.py330 def to_tensor(self, value): member in MaskedTensor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
H A Dxla_sharding.py281 def copy_sharding(from_tensor, to_tensor, use_sharding_op=False): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dconfig.proto835 string to_tensor = 2; field
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dautotune_process.py459 def to_tensor(self) -> torch.Tensor: member in TensorMeta
/aosp_15_r20/external/executorch/exir/
H A Dpass_base.py140 def to_tensor(self) -> torch.Tensor: member in ProxyValue
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dfunction_optimizer.cc336 SafeTensorId to_tensor(func_node, to_idx); in AddTensorMapping() local
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dutils.py1399 def to_tensor(t): function
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Dconfig.pb.h7999 inline const std::string& TensorConnection::to_tensor() const { in to_tensor() function
8030 inline void TensorConnection::set_allocated_to_tensor(std::string* to_tensor) { in set_allocated_to_tensor()
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_tracer.py549 def to_tensor(x, y): function