Home
last modified time | relevance | path

Searched refs:to_tensor (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_rma_distributed.cc38 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 …]
Dcollective_rma_distributed.h38 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor,
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_collective_executor_mgr.cc39 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/
Dcollective_rma_local.cc29 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()
Dbase_collective_executor.h124 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument
129 to_alloc_attr, to_tensor, client_locality, stream_index, done); in RecvFromPeer()
Dgraph_execution_state.cc518 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()
Dcollective_rma_local.h42 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor,
Dtest_collective_executor_mgr.h34 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument
Dring_gatherer_test.cc69 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()
Dring_reducer_test.cc69 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()
Dhierarchical_tree_broadcaster_test.cc165 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/
Dragged_to_tensor_op_test.py38 dt = rt.to_tensor()
104 dt = rt.to_tensor(default)
134 rt.to_tensor(default)
Dragged_conversion_ops.py32 def to_tensor(rt_input, default_value=None, name=None): function
34 return rt_input.to_tensor(default_value, name)
Dragged_gather_ops.py204 result = ragged_conversion_ops.to_tensor(result)
Dragged_tensor.py1193 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/
Dcomposite_tensor_support_test.py48 return inputs.to_tensor(default_value=self._default_value)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-ragged-tensor.pbtxt115 name: "to_tensor"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-ragged-tensor.pbtxt115 name: "to_tensor"
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc374 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/
Dcollective.h239 Tensor* to_tensor,
/external/tensorflow/tensorflow/python/kernel_tests/
Dunicode_decode_op_test.py227 input_tensor, "UTF-8").to_tensor(default_value=-1)
575 input_tensor, "UTF-8").to_tensor(default_value="")
/external/tensorflow/tensorflow/core/protobuf/
Dconfig.proto547 // the tensor named in `to_tensor`.
552 string to_tensor = 2; field
/external/tensorflow/tensorflow/python/keras/
Dbackend.py3034 connection.to_tensor = x.name # Placeholder