/external/tensorflow/tensorflow/core/distributed_runtime/ |
D | collective_rma_distributed.cc | 38 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvBufCall() argument 47 req_.set_num_bytes(to_tensor->TotalBytes()); in RecvBufCall() 48 req_.set_buf_ptr(reinterpret_cast<int64>(DMAHelper::base(to_tensor))); in RecvBufCall() 78 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument 84 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer() 99 to_device_ctx, to_tensor, dev_to_dev_stream_index, in RecvFromPeer() 111 if (num_bytes != to_tensor->TotalBytes()) { in RecvFromPeer() 114 to_tensor->TotalBytes())); in RecvFromPeer() 132 to_tensor->dtype(), to_tensor->shape()); in RecvFromPeer() 138 to_tensor, dev_to_dev_stream_index, in RecvFromPeer() [all …]
|
D | collective_rma_distributed.h | 38 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor,
|
/external/tensorflow/tensorflow/contrib/gdr/ |
D | gdr_collective_executor_mgr.cc | 39 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvBufCall() argument 48 req_.set_num_bytes(to_tensor->TotalBytes()); in RecvBufCall() 49 req_.set_buf_ptr(reinterpret_cast<int64>(DMAHelper::base(to_tensor))); in RecvBufCall() 81 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument 88 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer() 103 to_device_ctx, to_tensor, done](const Status& s) { in RecvFromPeer() 106 to_tensor, state->call->resp_.transport_options(), to_device, in RecvFromPeer() 120 to_tensor, client_locality, in RecvFromPeer() 127 to_alloc_attr, to_tensor, client_locality, state->server_locality, in RecvFromPeer()
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | collective_rma_local.cc | 29 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument 41 key, [to_tensor, to_device_ctx, to_device, to_alloc_attr, in RecvFromPeer() 48 int64 recv_bytes = to_tensor->TotalBytes(); in RecvFromPeer() 57 to_tensor, // dst Tensor* in RecvFromPeer()
|
D | base_collective_executor.h | 124 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument 129 to_alloc_attr, to_tensor, client_locality, stream_index, done); in RecvFromPeer()
|
D | graph_execution_state.cc | 518 tensor_connection.to_tensor(), "\"."); in PruneGraph() 523 tensor_connection.to_tensor(), in PruneGraph() 528 &tensor_connection.to_tensor(), {from_node, from_id.second})); in PruneGraph() 643 TensorId id = ParseTensorName(tensor_connection.to_tensor()); in OptimizeGraph() 646 tensor_connection.to_tensor()); in OptimizeGraph()
|
D | collective_rma_local.h | 42 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor,
|
D | test_collective_executor_mgr.h | 34 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument
|
D | ring_gatherer_test.cc | 69 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument 76 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
|
D | ring_reducer_test.cc | 69 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument 76 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
|
D | hierarchical_tree_broadcaster_test.cc | 165 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument 171 to_alloc_attr, to_tensor, client_locality, stream_index, done); in RecvFromPeer()
|
/external/tensorflow/tensorflow/python/ops/ragged/ |
D | ragged_to_tensor_op_test.py | 38 dt = rt.to_tensor() 104 dt = rt.to_tensor(default) 134 rt.to_tensor(default)
|
D | ragged_conversion_ops.py | 32 def to_tensor(rt_input, default_value=None, name=None): function 34 return rt_input.to_tensor(default_value, name)
|
D | ragged_gather_ops.py | 204 result = ragged_conversion_ops.to_tensor(result)
|
D | ragged_tensor.py | 1193 dense_ragged_mask = ragged_mask.to_tensor(default_value=False) 1222 def to_tensor(self, default_value=None, name=None): member in RaggedTensor 1256 values = values.to_tensor(default_value)
|
/external/tensorflow/tensorflow/python/keras/utils/ |
D | composite_tensor_support_test.py | 48 return inputs.to_tensor(default_value=self._default_value)
|
/external/tensorflow/tensorflow/tools/api/golden/v2/ |
D | tensorflow.-ragged-tensor.pbtxt | 115 name: "to_tensor"
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.-ragged-tensor.pbtxt | 115 name: "to_tensor"
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | function_optimizer.cc | 374 SafeTensorId to_tensor(func_node, to_idx); in AddTensorMapping() local 375 AddTensorMapping(from_tensor, to_tensor); in AddTensorMapping() 1904 const SafeTensorId to_tensor = ParseTensorName(output_tensor); in InlineIndirectFunctionCall() local 1907 SafeTensorId(absl::StrCat(func_node.name(), "/", to_tensor.node()), in InlineIndirectFunctionCall() 1908 to_tensor.index()); in InlineIndirectFunctionCall()
|
/external/tensorflow/tensorflow/core/framework/ |
D | collective.h | 239 Tensor* to_tensor,
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | unicode_decode_op_test.py | 227 input_tensor, "UTF-8").to_tensor(default_value=-1) 575 input_tensor, "UTF-8").to_tensor(default_value="")
|
/external/tensorflow/tensorflow/core/protobuf/ |
D | config.proto | 547 // the tensor named in `to_tensor`. 552 string to_tensor = 2; field
|
/external/tensorflow/tensorflow/python/keras/ |
D | backend.py | 3034 connection.to_tensor = x.name # Placeholder
|