| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_casts.cpp | 45 Tensor<double, 2> dtensor(20,30); in test_vectorized_cast() local
64 Tensor<double, 2> dtensor(20,30); in test_float_to_int_cast() local
81 Tensor<double, 2> dtensor(20, 30); in test_big_to_small_type_cast() local
98 Tensor<double, 2> dtensor(20, 30); in test_small_to_big_type_cast() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | sparse_expanders.cc | 22 namespace dtensor { namespace
|
| D | spmd_expanders.cc | 58 namespace dtensor { namespace
|
| D | dce.cc | 26 namespace dtensor { namespace
|
| D | device_utils.cc | 23 namespace dtensor { namespace
|
| D | collectives_common.cc | 21 namespace dtensor { namespace
|
| D | function_renaming.cc | 29 namespace dtensor { namespace
|
| D | op_utils.cc | 23 namespace dtensor { namespace
|
| D | sparse_expander.cc | 27 namespace dtensor { namespace
|
| D | undo_merge_const_across_mesh.cc | 32 namespace dtensor { namespace
|
| D | set_default_sharding.cc | 31 namespace dtensor { namespace
|
| D | constant_folding.cc | 26 namespace dtensor { namespace
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | tpu_system_interface.cc | 19 namespace dtensor { namespace
|
| D | dtensor_tpu_ops.cc | 27 namespace dtensor { namespace
|
| D | dtensor_ops.cc | 29 namespace dtensor { namespace
|
| /external/tensorflow/tensorflow/dtensor/mlir/sparse_expansions/ |
| D | matmul_sparse_expander.cc | 23 namespace dtensor { namespace
|
| /external/tensorflow/tensorflow/dtensor/mlir/utils/ |
| D | dtensor_embedding_stub.cc | 20 namespace dtensor { namespace
|
| D | dtensor_embedding_v2_stub.cc | 20 namespace dtensor { namespace
|
| D | dtensor_mlir_passes_internal.cc | 22 namespace dtensor { namespace
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | list_ops.py | 232 def _TensorListStackGrad(unused_op, dtensor): argument
238 def _TensorListConcatGrad(op, dtensor, unused_dlengths): argument
324 def _TensorListGatherGrad(op, dtensor): argument
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | tensorlist_setitem_spmd_expander.cc | 24 namespace dtensor { namespace
|
| D | tensorlist_getitem_spmd_expander.cc | 23 namespace dtensor { namespace
|
| D | range_spmd_expander.cc | 24 namespace dtensor { namespace
|
| D | sparse_to_dense_spmd_expander.cc | 23 namespace dtensor { namespace
|
| D | tensorlist_reserve_spmd_expander.cc | 27 namespace dtensor { namespace
|