Searched refs:index_t (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | random_index_shuffle_ops.cc | 71 const Tensor& index_t = context->input(0); in Compute() local
76 index_t.dims() == 0 && seed_t.dims() == 1 && max_index_t.dims() == 0; in Compute()
78 std::max(std::max(index_t.NumElements(), max_index_t.NumElements()), in Compute()
83 index_t.dims() == 0 || in Compute()
84 (index_t.dims() == 1 && index_t.dim_size(0) == num_outputs), in Compute()
107 context, context->allocate_output(0, index_t.shape(), &new_index_t)); in Compute()
116 static_cast<uint64_t>(index_t.dims() ? index_t.vec<IntType>()(i) in Compute()
117 : index_t.scalar<IntType>()()); in Compute()
|
| D | tensor_array.h | 550 size_t index_t = static_cast<size_t>(index); in LockedRead() local
551 if ((is_grad_ && (index_t >= tensors_.size() || !tensors_[index].written)) || in LockedRead()
552 (!is_grad_ && (index_t < tensors_.size() && !tensors_[index].written))) { in LockedRead()
558 if (is_grad_ && index_t < tensors_.size() && in LockedRead()
579 if (index_t >= tensors_.size()) { in LockedRead()
|
| D | gather_nd_op.cc | 37 const DataType index_t = DataTypeToEnum<Index>::v(); in GatherNdOp() local
38 OP_REQUIRES_OK(c, c->MatchSignature({dt, index_t}, {dt})); in GatherNdOp()
|
| D | scatter_nd_op.cc | 66 const DataType index_t = DataTypeToEnum<Index>::v(); in ScatterNdOp() local
67 OP_REQUIRES_OK(c, c->MatchSignature({index_t, dt, index_t}, {dt})); in ScatterNdOp()
142 const DataType index_t = DataTypeToEnum<Index>::v(); in TensorScatterOp() local
143 OP_REQUIRES_OK(c, c->MatchSignature({dt, index_t, dt}, {dt})); in TensorScatterOp()
237 const DataType index_t = DataTypeToEnum<Index>::v(); in ScatterNdUpdateOp() local
245 OP_REQUIRES_OK(c, c->MatchSignature({dt_ref, index_t, dt}, {dt_ref})); in ScatterNdUpdateOp()
248 OP_REQUIRES_OK(c, c->MatchSignature({dt, index_t, dt}, {dt})); in ScatterNdUpdateOp()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorScanSycl.h | 48 template <typename index_t>
51 static EIGEN_CONSTEXPR index_t ScanPerThread = 8;
52 const index_t total_size;
53 const index_t non_scan_size;
54 const index_t scan_size;
55 const index_t non_scan_stride;
56 const index_t scan_stride;
57 const index_t panel_threads;
58 const index_t group_threads;
59 const index_t block_threads;
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_matrix_components_op.cc | 51 const Tensor& index_t = c->input(1); in Compute() local
57 OP_REQUIRES(c, index_t.dims() == 0, in Compute()
59 index_t.DebugString())); in Compute()
60 int32_t index = index_t.scalar<int32>()(); in Compute()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | batch_ops_test.py | 79 index_t = thread_results[1]
84 index_t = main_results[1]
91 self.assertEqual(len(index_t), 2)
|
| /external/oss-fuzz/projects/alembic/ |
| D | alembic_dump_info_fuzzer.cc | 44 using Alembic::AbcGeom::index_t;
|
| /external/tensorflow/tensorflow/python/kernel_tests/data_structures/ |
| D | barrier_ops_test.py | 150 index_t, key_t, value_list_t = b.take_many(3, allow_small_batch=True)
160 [index_t, key_t, value_list_t[0], value_list_t[1]])
174 [index_t, key_t, value_list_t[0], value_list_t[1]])
|
| /external/jemalloc_new/ |
| D | ChangeLog | 676 + Rename index_t to szind_t to avoid an existing type on Solaris.
|