Searched refs:dtensor (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_casts.cpp | 44 Tensor<double, 2> dtensor(20,30); in test_vectorized_cast() local
45 dtensor.setRandom(); in test_vectorized_cast()
48 dtensor = itensor.cast<double>(); in test_vectorized_cast()
53 VERIFY_IS_EQUAL(dtensor(i,j), static_cast<double>(ftensor(i,j))); in test_vectorized_cast()
63 Tensor<double, 2> dtensor(20,30); in test_float_to_int_cast() local
64 dtensor = dtensor.random() * 1000.0; in test_float_to_int_cast()
67 Tensor<int, 2> i2tensor = dtensor.cast<int>(); in test_float_to_int_cast()
72 VERIFY_IS_EQUAL(i2tensor(i,j), static_cast<int>(dtensor(i,j))); in test_float_to_int_cast()
80 Tensor<double, 2> dtensor(20, 30); in test_big_to_small_type_cast() local
81 dtensor.setRandom(); in test_big_to_small_type_cast()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | list_ops.py | 193 def _TensorListStackGrad(unused_op, dtensor): argument
194 return tensor_list_from_tensor(dtensor, element_shape=dtensor.shape[1:]), None
199 def _TensorListConcatGrad(op, dtensor, unused_dlengths): argument
202 dtensor,
285 def _TensorListGatherGrad(op, dtensor): argument
291 dlist = tensor_list_reserve(element_shape, num_elements, dtensor.dtype)
293 tensor=dtensor, indices=indices, input_handle=dlist)
303 dtensor = gen_list_ops.tensor_list_gather(
309 return dtensor, None, None, None
311 return dtensor, None, None
[all …]
|