| /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 indices_t->dim_size(0) < std::numeric_limits<int>::max(), in Compute()
70 const int nnz = static_cast<int>(indices_t->dim_size(0)); in Compute()
71 const int rank = static_cast<int>(indices_t->dim_size(1)); in Compute()
75 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; in Compute() local
47 OP_REQUIRES_OK(context, context->input("indices", &indices_t)); in Compute()
57 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(indices_t->shape()), in Compute()
59 indices_t->shape().DebugString())); in Compute()
71 const auto indices = indices_t->matrix<int64>(); in Compute()
75 const int64 N = indices_t->shape().dim_size(0); in Compute()
89 int rank = indices_t->shape().dim_size(1); 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.cc | 109 auto indices_t = in Compute() local
117 indices_t, on_value_t, in Compute()
|
| D | ctc_decoder_ops.cc | 144 auto indices_t = p_indices->matrix<int64>(); in StoreAllDecodedSequences() local
157 indices_t(offset, 0) = b; in StoreAllDecodedSequences()
158 indices_t(offset, 1) = t; in StoreAllDecodedSequences()
|
| D | tensor_array_ops.cc | 633 const auto indices_t = tensor_indices->vec<int32>(); in Compute() local
636 std::copy(indices_t.data(), indices_t.data() + num_indices, in Compute()
1087 const auto indices_t = tensor_indices->vec<int32>(); in Compute() local
1091 : *std::max_element(indices_t.data(), in Compute()
1092 indices_t.data() + num_values); in Compute()
1095 std::copy(indices_t.data(), indices_t.data() + num_values, in Compute()
|
| D | example_parsing_ops.cc | 692 auto indices_t = sp_indices_d->matrix<int64>(); in Compute() local
693 std::iota(indices_t.data(), indices_t.data() + num_elements, 0); in Compute()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops_test.py | 75 indices_t = constant_op.constant(indices)
78 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 | 2873 Tensor indices_t(DT_INT32, TensorShape({2})); in TEST_F() local
2874 indices_t.flat<int>()(0) = 0; in TEST_F()
2875 indices_t.flat<int>()(1) = 1; in TEST_F()
2877 item.graph, item.fetch, {{"input", input_t}, {"indices", indices_t}}); in TEST_F()
2912 {{"input", input_t}, {"indices", indices_t}}); in TEST_F()
|