Home
last modified time | relevance | path

Searched refs:dtensor (Results 1 – 6 of 6) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_casts.cpp44 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/
Dlist_ops.py235 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/
DBUILD7 "//learning/brain/experimental/dtensor:__subpackages__",
/external/tensorflow/tensorflow/compiler/mlir/xla/
DBUILD16 "//learning/brain/experimental/dtensor/...",
/external/tensorflow/tensorflow/compiler/mlir/hlo/
DBUILD23 "//learning/brain/experimental/dtensor/...",
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/
DBUILD19 "//learning/brain/experimental/dtensor/...",