Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_rma_distributed.cc40 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvBufCall()
81 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_conversion_ops.py48 def to_tensor(rt_input, default_value=None, name=None): function
Dragged_tensor.py1701 def to_tensor(self, default_value=None, name=None, shape=None): member in RaggedTensor
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_rma_local.cc29 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
Dpermuter_test.cc78 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
Dring_gatherer_test.cc70 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
Dring_reducer_test.cc71 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
Dhierarchical_tree_broadcaster_test.cc168 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
Dxla_sharding.py234 def copy_sharding(from_tensor, to_tensor, use_sharding_op=False): argument
/external/tensorflow/tensorflow/core/protobuf/
Dconfig.proto746 string to_tensor = 2; field
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc333 SafeTensorId to_tensor(func_node, to_idx); in AddTensorMapping() local