| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_slice_grad_op.cc | 30 const Tensor *backprop_val_grad, *input_indices, *output_indices, *input_start; in Compute() local
32 OP_REQUIRES_OK(ctx, ctx->input("input_indices", &input_indices)); in Compute()
37 TensorShapeUtils::IsMatrix(input_indices->shape()) && in Compute()
42 input_indices->shape().DebugString(), " and ", in Compute()
51 input_indices->dim_size(1) == output_indices->dim_size(1), in Compute()
53 "ndims: got: ", input_indices->dim_size(1), " and ", in Compute()
56 ctx, output_indices->dim_size(0) <= input_indices->dim_size(0), in Compute()
60 input_indices->dim_size(0))); in Compute()
72 const int num_dims = input_indices->dim_size(1); in Compute()
78 const int64 input_nnz = input_indices->dim_size(0); in Compute()
[all …]
|
| D | dynamic_stitch_op_gpu.cu.cc | 34 GpuDeviceArrayStruct<int32> input_indices, in DynamicStitchKernel() argument
37 int32* data_indices = GetGpuDeviceArrayOnDevice(&input_indices); in DynamicStitchKernel()
54 const GpuDeviceArrayStruct<int>& input_indices, in DynamicStitchGPUImpl() argument
62 slice_size, output_size, input_indices, in DynamicStitchGPUImpl()
70 const GpuDeviceArrayStruct<int32>& input_indices, \
|
| D | serialize_sparse_op.cc | 55 const Tensor* input_indices; in Compute() local
59 OP_REQUIRES_OK(context, context->input("sparse_indices", &input_indices)); in Compute()
62 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices->shape()), in Compute()
65 input_indices->shape().DebugString())); in Compute()
81 OP_REQUIRES_OK(context, Serialize(*input_indices, &serialized_sparse_t(0))); in Compute()
162 const Tensor* input_indices; in Compute() local
165 OP_REQUIRES_OK(context, context->input("sparse_indices", &input_indices)); in Compute()
168 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices->shape()), in Compute()
171 input_indices->shape().DebugString())); in Compute()
194 OP_REQUIRES_OK(context, SparseTensor::Create(*input_indices, *input_values, in Compute()
|
| D | sparse_split_op.cc | 34 const Tensor& input_indices = context->input(1); in Compute() local
38 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices.shape()), in Compute()
41 input_indices.shape().DebugString())); in Compute()
45 input_indices.shape().DebugString())); in Compute()
69 input_indices, input_values, in Compute()
|
| D | sparse_slice_op.cc | 31 const Tensor& input_indices = context->input(0); in Compute() local
37 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices.shape()), in Compute()
40 input_indices.shape().DebugString())); in Compute()
44 input_indices.shape().DebugString())); in Compute()
72 input_indices, input_values, in Compute()
|
| D | aggregate_ops.cc | 59 gtl::InlinedVector<int, 8> input_indices(num); in Compute() local
60 std::iota(input_indices.begin(), input_indices.end(), 0); in Compute()
74 input_indices[0] = reused_input; in Compute()
75 input_indices[reused_input] = 0; in Compute()
79 #define I(IDX) ctx->input(input_indices[IDX]).flat<T>() in Compute()
|
| D | sparse_tensors_map_ops.cc | 169 const Tensor* input_indices; in Compute() local
174 OP_REQUIRES_OK(context, context->input("sparse_indices", &input_indices)); in Compute()
179 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices->shape()), in Compute()
182 input_indices->shape().DebugString())); in Compute()
200 OP_REQUIRES_OK(context, SparseTensor::Create(*input_indices, *input_values, in Compute()
224 const Tensor* input_indices; in Compute() local
229 OP_REQUIRES_OK(context, context->input("sparse_indices", &input_indices)); in Compute()
234 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices->shape()), in Compute()
237 input_indices->shape().DebugString())); in Compute()
260 OP_REQUIRES_OK(context, SparseTensor::Create(*input_indices, *input_values, in Compute()
|
| D | topk_op_gpu.h | 443 Tensor input_indices;
445 DT_INT32, TensorShape({num_rows, num_cols}), &input_indices));
446 auto input_indices_t = To32Bit(input_indices.flat<int32>());
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseReorder.pbtxt | 4 name: "input_indices"
13 1-D. `N` non-empty values corresponding to `input_indices`.
25 2-D. `N x R` matrix with the same indices as input_indices, but
43 If the tensor has rank `R` and `N` non-empty values, `input_indices` has
|
| D | api_def_SparseReshape.pbtxt | 4 name: "input_indices"
40 tensor. The `input_indices` are recomputed based on the requested `new_shape`.
51 has length `R_out`, then `input_indices` has shape `[N, R_in]`,
|
| D | api_def_SparseReduceSumSparse.pbtxt | 4 name: "input_indices"
13 1-D. `N` non-empty values corresponding to `input_indices`.
|
| D | api_def_SparseReduceMaxSparse.pbtxt | 4 name: "input_indices"
13 1-D. `N` non-empty values corresponding to `input_indices`.
|
| D | api_def_SparseReduceMax.pbtxt | 4 name: "input_indices"
13 1-D. `N` non-empty values corresponding to `input_indices`.
|
| D | api_def_SparseReduceSum.pbtxt | 4 name: "input_indices"
13 1-D. `N` non-empty values corresponding to `input_indices`.
|
| D | api_def_SparseSliceGrad.pbtxt | 11 name: "input_indices"
|
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| D | wals_solver_ops.cc | 77 const Tensor& input_indices = context->input(4); in Compute() local
92 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices.shape()), in Compute()
95 context, input_indices.dim_size(1) == 2, in Compute()
101 OP_REQUIRES(context, input_indices.dim_size(0) == input_values.dim_size(0), in Compute()
115 entry_weights.dim_size(0) == input_indices.dim_size(0))), in Compute()
124 const int64 num_nonzero_elements = input_indices.dim_size(0); in Compute()
134 ConstEigenMatrixInt64Map indices_mat(input_indices.matrix<int64>().data(), in Compute()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_grad.py | 50 input_indices = op.inputs[0]
53 num_entries = array_ops.shape(input_indices)[0]
56 input_indices, entry_indices, input_shape)
137 input_indices = op.inputs[0]
142 backprop_val_grad, input_indices, input_start, output_indices)
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | graph_jni.cc | 214 jintArray input_indices = env->NewIntArray(ninputs); in buildSubgraph() local
220 jint* input_indices_elems = env->GetIntArrayElements(input_indices, nullptr); in buildSubgraph()
237 env->ReleaseIntArrayElements(input_indices, input_indices_elems, 0); in buildSubgraph()
245 input_handles, input_indices, output_handles, output_indices); in buildSubgraph()
258 jintArray input_indices, jstring name, jobject cond_graph_builder, in Java_org_tensorflow_Graph_whileLoop() argument
267 resolveOutputs(env, "inputs", input_handles, input_indices, inputs.get(), in Java_org_tensorflow_Graph_whileLoop()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | reorder_reshape_transpose.cc | 62 std::vector<int> input_indices; in ComputeNewPerm() local
65 input_indices.push_back(i); in ComputeNewPerm()
75 input_indices[intermediate_to_input_indices_map.size()]; in ComputeNewPerm()
|
| D | resolve_constant_unary.cc | 66 std::vector<int> input_indices = ReverseOffset(input_shape, input_offset); in ReduceGeneric() local
70 output_indices[i] = input_indices[i] * reduction_mask[i]; in ReduceGeneric()
73 if (input_indices == output_indices) { in ReduceGeneric()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 569 ShapeHandle input_indices = c->input(0); in __anona7ffb82a1602() local
570 TF_RETURN_IF_ERROR(c->WithRank(input_indices, 2, &input_indices)); in __anona7ffb82a1602()
577 DimensionHandle N = c->Dim(input_indices, 0); in __anona7ffb82a1602()
580 TF_RETURN_IF_ERROR(c->Merge(c->Dim(input_indices, 1), in __anona7ffb82a1602()
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | wals.py | 259 input_indices = sp_input.indices[:, 0]
267 input_indices,
268 array_ops.ones_like(input_indices, dtype=dtypes.bool),
|
| /external/libtextclassifier/annotator/ |
| D | annotator_jni.cc | 530 CodepointSpan input_indices = in TC3_JNI_METHOD() local
533 context_utf8, input_indices, FromJavaSelectionOptions(env, options)); in TC3_JNI_METHOD()
553 const CodepointSpan input_indices = in TC3_JNI_METHOD() local
558 model_context->model()->ClassifyText(context_utf8, input_indices, in TC3_JNI_METHOD()
563 &classification_options, context_utf8, input_indices, in TC3_JNI_METHOD()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen.cc | 163 string FlattenInputs(const std::vector<int>* input_indices,
185 const std::vector<int>* input_indices, in FlattenInputs() argument
189 const int n = input_indices != nullptr ? input_indices->size() in FlattenInputs()
192 const int i = input_indices ? (*input_indices)[j] : j; in FlattenInputs()
|
| /external/tensorflow/tensorflow/contrib/lookup/ |
| D | lookup_ops_test.py | 1354 input_indices = [[0, 0], [0, 1], [2, 0], [2, 2], [3, 0]]
1358 constant_op.constant(input_indices, dtypes.int64),
1378 self.assertAllEqual(input_indices, sp_ids_ind)
1383 input_indices = [[0, 0], [0, 1], [2, 0], [2, 2], [3, 0]]
1387 constant_op.constant(input_indices, dtypes.int64),
1407 self.assertAllEqual(input_indices, sp_ids_ind)
1412 input_indices = [[0, 0], [0, 1], [2, 0], [2, 2], [3, 0]]
1416 constant_op.constant(input_indices, dtypes.int64),
1436 self.assertAllEqual(input_indices, sp_ids_ind)
|