| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_dense_binary_op_shared.cc | 62 const Tensor *indices_t, *values_t, *shape_t, *dense_t; variable
63 OP_REQUIRES_OK(ctx, ctx->input("sp_indices", &indices_t));
69 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(indices_t->shape()),
72 indices_t->shape().DebugString()));
81 OP_REQUIRES(ctx, indices_t->dim_size(0) < std::numeric_limits<int>::max(),
85 const auto indices_mat = indices_t->matrix<int64>();
109 const int nnz = static_cast<int>(indices_t->dim_size(0));
|
| sparse_softmax_op.cc | 43 const Tensor *indices_t, *values_t, *shape_t; variable
44 OP_REQUIRES_OK(context, context->input("sp_indices", &indices_t));
49 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(indices_t->shape()),
52 indices_t->shape().DebugString()));
66 indices_t->dim_size(0) < std::numeric_limits<int>::max(),
70 const int nnz = static_cast<int>(indices_t->dim_size(0));
71 const int rank = static_cast<int>(indices_t->dim_size(1));
72 SparseTensor st(tensor::DeepCopy(*indices_t), tensor::DeepCopy(*values_t),
|
| one_hot_op.cc | 109 auto indices_t = variable
117 indices_t, on_value_t,
|
| sparse_fill_empty_rows_op.cc | 43 const Tensor* indices_t; variable
47 OP_REQUIRES_OK(context, context->input("indices", &indices_t));
57 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(indices_t->shape()),
59 indices_t->shape().DebugString()));
71 const auto indices = indices_t->matrix<int64>();
75 const int64 N = indices_t->shape().dim_size(0);
89 int rank = indices_t->shape().dim_size(1);
|
| sparse_reduce_op.cc | 161 const Tensor *indices_t, *values_t, *shape_t, *reduction_axes_t; variable
162 OP_REQUIRES_OK(ctx, ctx->input("input_indices", &indices_t));
175 SparseTensor sp(tensor::DeepCopy(*indices_t), tensor::DeepCopy(*values_t),
255 const Tensor *indices_t, *values_t, *shape_t, *reduction_axes_t; variable
256 OP_REQUIRES_OK(ctx, ctx->input("input_indices", &indices_t));
263 SparseTensor sp(tensor::DeepCopy(*indices_t), tensor::DeepCopy(*values_t),
|
| example_parsing_ops.cc | 518 auto indices_t = sp_indices_d->matrix<int64>(); variable
519 std::iota(indices_t.data(), indices_t.data() + num_elements, 0);
|
| tensor_array_ops.cc | 600 const auto indices_t = tensor_indices->vec<int32>(); variable
603 std::copy(indices_t.data(), indices_t.data() + num_indices,
1053 const auto indices_t = tensor_indices->vec<int32>(); variable
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| data_flow_ops.cc | 99 const Tensor* indices_t = c->input_tensor(i); local
100 if (indices_t == nullptr) {
121 if (indices_t != nullptr) {
123 const int32* indices = indices_t->flat<int32>().data();
124 int64 count = indices_t->NumElements();
[all...] |
