/external/tensorflow/tensorflow/core/distributed_runtime/ |
D | collective_rma_distributed.cc | 40 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvBufCall() argument 49 req_.set_num_bytes(to_tensor->TotalBytes()); in RecvBufCall() 50 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 87 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer() 102 to_device_ctx, to_tensor, dev_to_dev_stream_index, in RecvFromPeer() 114 const int64 total_bytes = to_tensor->TotalBytes(); in RecvFromPeer() 118 to_tensor->TotalBytes())); in RecvFromPeer() 138 step_id_, "dynamic", to_tensor->dtype(), &to_tensor->shape()); in RecvFromPeer() 140 to_tensor->dtype(), to_tensor->shape()); in RecvFromPeer() [all …]
|
D | collective_rma_distributed.h | 43 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor,
|
/external/tensorflow/tensorflow/python/ops/ragged/ |
D | ragged_to_tensor_op_test.py | 97 dt = rt.to_tensor() 341 dt = rt.to_tensor(default_value=default, shape=shape) 400 self.evaluate(rt.to_tensor(default_value=default, shape=shape)) 405 rt_placeholder.to_tensor(default_value=default, shape=shape)) 409 actual = input_data.to_tensor( 416 actual = input_data.to_tensor( 424 actual = input_data.to_tensor( 431 actual = input_data.to_tensor( 438 actual = input_data.to_tensor(shape=[4, 4]) 460 actual = input_data.to_tensor(default_value=default_value) [all …]
|
D | ragged_squeeze_op_test.py | 54 self.assertAllEqual(ragged_conversion_ops.to_tensor(rt), dt) 114 self.assertAllEqual(ragged_conversion_ops.to_tensor(rt), dt) 169 self.assertAllEqual(ragged_conversion_ops.to_tensor(rt_s), dt_s)
|
D | ragged_conversion_ops.py | 48 def to_tensor(rt_input, default_value=None, name=None): function 50 return rt_input.to_tensor(default_value, name) 57 return rt_input.to_tensor(default_value=default_value, shape=shape)
|
D | ragged_string_ops.py | 768 to_tensor = False 771 to_tensor = True 792 dense_shape) if to_tensor else output 823 dense_shape) if to_tensor else output
|
D | ragged_one_hot_op_test.py | 194 self.assertAllEqual(result.to_tensor(), expected)
|
D | ragged_gather_op_test.py | 387 actual = actual.to_tensor()
|
/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/ |
D | xla_sharding.py | 234 def copy_sharding(from_tensor, to_tensor, use_sharding_op=False): argument 247 return to_tensor 250 to_tensor = tf2xla.sharding(to_tensor, sharding=sharding) 253 to_tensor.op._set_attr('_XlaSharding', attr_value) 254 return to_tensor
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | collective_rma_local.cc | 29 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument 48 auto consumer_callback = [to_tensor, to_device_ctx, to_device, to_alloc_attr, in RecvFromPeer() 65 int64 recv_bytes = to_tensor->TotalBytes(); in RecvFromPeer() 74 to_tensor, // dst Tensor* in RecvFromPeer()
|
D | collective_rma_local.h | 43 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor,
|
D | graph_execution_state.cc | 577 tensor_connection.to_tensor(), "\"."); in PruneGraph() 582 tensor_connection.to_tensor(), in PruneGraph() 587 &tensor_connection.to_tensor(), {from_node, from_id.second})); in PruneGraph() 690 feeds.emplace_back(ParseTensorName(tensor_connection.to_tensor())); in OptimizeGraph()
|
D | permuter_test.cc | 78 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument 85 to_alloc_attr, to_tensor, client_locality, stream_index, in RecvFromPeer()
|
D | ring_gatherer_test.cc | 70 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument 78 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
|
/external/tflite-support/tensorflow_lite_support/custom_ops/python/ |
D | ragged_tensor_to_tensor_test.py | 33 return ragged_tensor.to_tensor()
|
/external/tensorflow/tensorflow/python/keras/layers/ |
D | pooling_test.py | 65 dense_data = ragged_data.to_tensor() 85 dense_data = ragged_data.to_tensor()
|
D | merge_test.py | 241 dense_data = ragged_data.to_tensor() 247 out_ragged = convert_ragged_tensor_value(out_ragged).to_tensor()
|
/external/tensorflow/tensorflow/python/distribute/ |
D | input_lib_type_spec_test.py | 352 "dense": ragged_tensor.to_tensor(), 413 "dense": ragged_tensor.to_tensor(), 475 "dense": ragged_tensor.to_tensor(),
|
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
D | ngrams_test.py | 112 input_tensor = tf.ragged.constant(test_case).to_tensor() 132 input_tensor = tf.ragged.constant(test_case).to_tensor()
|
D | whitespace_tokenizer_test.py | 54 return tokenizer.tokenize(test_case).to_tensor()
|
/external/tensorflow/tensorflow/python/data/kernel_tests/ |
D | flat_map_test.py | 162 ragged_conversion_ops.to_tensor(x))
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.-ragged-tensor.pbtxt | 140 name: "to_tensor"
|
/external/tensorflow/tensorflow/tools/api/golden/v2/ |
D | tensorflow.-ragged-tensor.pbtxt | 140 name: "to_tensor"
|
/external/tensorflow/tensorflow/core/framework/ |
D | collective.h | 268 Tensor* to_tensor,
|
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
D | text_vectorization.py | 567 dense_data = lookup_data.to_tensor(default_value=0)
|