Home
last modified time | relevance | path

Searched refs:indices_t (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_softmax_op.cc43 const Tensor *indices_t, *values_t, *shape_t; in Compute() local
44 OP_REQUIRES_OK(context, context->input("sp_indices", &indices_t)); in Compute()
49 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(indices_t->shape()), in Compute()
52 indices_t->shape().DebugString())); in Compute()
66 const int64 nnz = indices_t->dim_size(0); in Compute()
67 const int rank = static_cast<int>(indices_t->dim_size(1)); in Compute()
71 tensor::DeepCopy(*indices_t), tensor::DeepCopy(*values_t), in Compute()
Dsparse_dense_binary_op_shared.cc62 const Tensor *indices_t, *values_t, *shape_t, *dense_t; in Compute() local
63 OP_REQUIRES_OK(ctx, ctx->input("sp_indices", &indices_t)); in Compute()
69 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(indices_t->shape()), in Compute()
72 indices_t->shape().DebugString())); in Compute()
82 const auto indices_mat = indices_t->matrix<int64>(); in Compute()
106 const int64 nnz = indices_t->dim_size(0); in Compute()
Dsparse_fill_empty_rows_op.cc43 const Tensor& indices_t, const Tensor& values_t, in operator ()()
51 const auto indices = indices_t.matrix<Tindex>(); in operator ()()
55 const Tindex N = indices_t.shape().dim_size(0); in operator ()()
74 int rank = indices_t.shape().dim_size(1); in operator ()()
130 context->set_output(kOutputIndicesOutput, indices_t); in operator ()()
200 const Tensor& indices_t = context->input(kIndicesInput); in Compute() local
208 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(indices_t.shape()), in Compute()
210 indices_t.shape().DebugString())); in Compute()
221 context, default_value_t, indices_t, values_t, in Compute()
Dsparse_reduce_op.cc161 const Tensor *indices_t, *values_t, *shape_t, *reduction_axes_t; in Compute() local
162 OP_REQUIRES_OK(ctx, ctx->input("input_indices", &indices_t)); in Compute()
177 tensor::DeepCopy(*indices_t), tensor::DeepCopy(*values_t), in Compute()
257 const Tensor *indices_t, *values_t, *shape_t, *reduction_axes_t; in Compute() local
258 OP_REQUIRES_OK(ctx, ctx->input("input_indices", &indices_t)); in Compute()
266 OP_REQUIRES_OK(ctx, SparseTensor::Create(tensor::DeepCopy(*indices_t), in Compute()
Done_hot_op_test.cc33 auto indices_t = indices.flat<int32>(); in OneHot() local
35 indices_t(i) = dist(gen); in OneHot()
Done_hot_op.cc110 auto indices_t = in Compute() local
118 indices_t, on_value_t, in Compute()
Dsparse_fill_empty_rows_op.h30 const Tensor& indices_t, const Tensor& values_t,
Dctc_decoder_ops.cc146 auto indices_t = p_indices->matrix<int64>(); in StoreAllDecodedSequences() local
166 indices_t(offset, 0) = b; in StoreAllDecodedSequences()
167 indices_t(offset, 1) = t; in StoreAllDecodedSequences()
Dtensor_array_ops.cc628 const auto indices_t = tensor_indices->vec<int32>(); in Compute() local
631 std::copy(indices_t.data(), indices_t.data() + num_indices, in Compute()
1080 const auto indices_t = tensor_indices->vec<int32>(); in Compute() local
1084 : *std::max_element(indices_t.data(), in Compute()
1085 indices_t.data() + num_values); in Compute()
1088 std::copy(indices_t.data(), indices_t.data() + num_values, in Compute()
Dragged_cross_op.cc126 SparseFeatureReader(const Tensor& indices_t, const Tensor& values_t, in SparseFeatureReader() argument
131 auto indices = indices_t.matrix<int64>(); in SparseFeatureReader()
Dexample_parsing_ops.cc1048 auto indices_t = sp_indices_d->matrix<int64>(); in Compute() local
1049 std::iota(indices_t.data(), indices_t.data() + num_elements, 0); in Compute()
/external/tensorflow/tensorflow/core/kernels/sparse/
Ddense_to_csr_sparse_matrix_op.cc160 const Tensor& indices_t = c->input(1); in ComputeAsync() local
169 c, indices_t.dims() == 2, in ComputeAsync()
171 indices_t.shape().DebugString()), in ComputeAsync()
174 c, indices_t.dim_size(1) == rank, in ComputeAsync()
177 indices_t.dim_size(1), " vs. ", rank), in ComputeAsync()
188 nnz_per_batch_host.mutable_data()[0] = indices_t.dim_size(0); in ComputeAsync()
198 auto indices = indices_t.matrix<int64>(); in ComputeAsync()
224 &indices_t, dense_tensor_shape, rows, cols, done]() { in ComputeAsync()
239 c, params_t, indices_t, &temp_values_t)), in ComputeAsync()
259 auto indices = indices_t.matrix<int64>(); in ComputeAsync()
Dsparse_tensor_to_csr_sparse_matrix_op.cc125 const Tensor& indices_t = c->input(0); in ComputeAsync() local
144 nnz_per_batch_host.mutable_data()[0] = indices_t.dim_size(0); in ComputeAsync()
154 auto indices = indices_t.matrix<int64>(); in ComputeAsync()
177 &indices_t, &dense_shape_t, dense_shape, rows, cols, in ComputeAsync()
192 auto indices = indices_t.matrix<int64>(); in ComputeAsync()
Dcsr_sparse_matrix_to_dense_op.cc150 Tensor indices_t; in Compute() local
152 &indices_t)); in Compute()
159 auto indices = indices_t.matrix<int64>(); in Compute()
201 c, indices_t, values_t, dense_tensor_shape, &dense_t, in Compute()
Dcsr_sparse_matrix_to_sparse_tensor_op.cc156 Tensor* indices_t; in Compute() local
158 c, c->allocate_output(0, TensorShape({total_nnz, rank}), &indices_t)); in Compute()
165 auto indices = indices_t->matrix<int64>(); in Compute()
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops_test.py98 indices_t = constant_op.constant(indices)
101 indices=indices_t, values=values_t, dense_shape=dense_shape_t)
/external/tensorflow/tensorflow/core/ops/
Ddata_flow_ops.cc99 const Tensor* indices_t = c->input_tensor(i); in DynamicStitchShapeFunction() local
100 if (indices_t == nullptr) { in DynamicStitchShapeFunction()
121 if (indices_t != nullptr) { in DynamicStitchShapeFunction()
123 const int32* indices = indices_t->flat<int32>().data(); in DynamicStitchShapeFunction()
124 int64 count = indices_t->NumElements(); in DynamicStitchShapeFunction()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding_test.cc3219 Tensor indices_t(DT_INT32, TensorShape({2})); in TEST_F() local
3220 indices_t.flat<int>()(0) = 0; in TEST_F()
3221 indices_t.flat<int>()(1) = 1; in TEST_F()
3223 item.graph, item.fetch, {{"input", input_t}, {"indices", indices_t}}); in TEST_F()
3258 {{"input", input_t}, {"indices", indices_t}}); in TEST_F()