Searched refs:dtensor (Results 1 – 6 of 6) 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 | 235 def _TensorListStackGrad(unused_op, dtensor): argument 236 return tensor_list_from_tensor(dtensor, element_shape=dtensor.shape[1:]), None 241 def _TensorListConcatGrad(op, dtensor, unused_dlengths): argument 244 dtensor, 327 def _TensorListGatherGrad(op, dtensor): argument 333 dlist = tensor_list_reserve(element_shape, num_elements, dtensor.dtype) 335 tensor=dtensor, indices=indices, input_handle=dlist) 345 dtensor = gen_list_ops.tensor_list_gather( 351 return dtensor, None, None, None 353 return dtensor, None, None [all …]
|
/external/tensorflow/tensorflow/stream_executor/tpu/ |
D | BUILD | 7 "//learning/brain/experimental/dtensor:__subpackages__",
|
/external/tensorflow/tensorflow/compiler/mlir/xla/ |
D | BUILD | 16 "//learning/brain/experimental/dtensor/...",
|
/external/tensorflow/tensorflow/compiler/mlir/hlo/ |
D | BUILD | 23 "//learning/brain/experimental/dtensor/...",
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ |
D | BUILD | 19 "//learning/brain/experimental/dtensor/...",
|