/external/tensorflow/tensorflow/core/kernels/ |
D | sparse_softmax_op.cc | 43 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()
|
D | sparse_dense_binary_op_shared.cc | 62 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()
|
D | sparse_fill_empty_rows_op.cc | 43 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()
|
D | sparse_reduce_op.cc | 161 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()
|
D | one_hot_op_test.cc | 33 auto indices_t = indices.flat<int32>(); in OneHot() local 35 indices_t(i) = dist(gen); in OneHot()
|
D | one_hot_op.cc | 110 auto indices_t = in Compute() local 118 indices_t, on_value_t, in Compute()
|
D | sparse_fill_empty_rows_op.h | 30 const Tensor& indices_t, const Tensor& values_t,
|
D | ctc_decoder_ops.cc | 146 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()
|
D | tensor_array_ops.cc | 628 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()
|
D | ragged_cross_op.cc | 126 SparseFeatureReader(const Tensor& indices_t, const Tensor& values_t, in SparseFeatureReader() argument 131 auto indices = indices_t.matrix<int64>(); in SparseFeatureReader()
|
D | example_parsing_ops.cc | 1048 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/ |
D | dense_to_csr_sparse_matrix_op.cc | 160 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()
|
D | sparse_tensor_to_csr_sparse_matrix_op.cc | 125 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()
|
D | csr_sparse_matrix_to_dense_op.cc | 150 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()
|
D | csr_sparse_matrix_to_sparse_tensor_op.cc | 156 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/ |
D | sparse_ops_test.py | 98 indices_t = constant_op.constant(indices) 101 indices=indices_t, values=values_t, dense_shape=dense_shape_t)
|
/external/tensorflow/tensorflow/core/ops/ |
D | data_flow_ops.cc | 99 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/ |
D | constant_folding_test.cc | 3219 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()
|