Home
last modified time | relevance | path

Searched refs:out_indices (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/contrib/libsvm/kernels/
Ddecode_libsvm_op.cc47 std::vector<std::pair<int64, int64>> out_indices; in Compute() local
87 out_indices.emplace_back(std::pair<int64, int64>(i, feature_index)); in Compute()
96 TensorShape({static_cast<int64>(out_indices.size()), in Compute()
107 for (int i = 0; i < out_indices.size(); i++) { in Compute()
108 indices(i, 0) = out_indices[i].first; in Compute()
109 int64 value = out_indices[i].first; in Compute()
114 indices(i, input_tensor->shape().dims()) = out_indices[i].second; in Compute()
/external/robolectric-shadows/resources/src/main/java/org/robolectric/res/android/
DAttributeResolution9.java92 int attrs_length, int[] out_values, int[] out_indices) { in ResolveAttrs() argument
223 if (out_indices != null && value.dataType != Res_value.TYPE_NULL) { in ResolveAttrs()
225 out_indices[indicesIdx] = ii; in ResolveAttrs()
231 if (out_indices != null) { in ResolveAttrs()
232 out_indices[0] = indicesIdx; in ResolveAttrs()
239 int[] out_values, int[] out_indices) { in ApplyStyle() argument
422 out_indices[indices_idx] = ii; in ApplyStyle()
441 out_indices[0] = indices_idx; in ApplyStyle()
445 int attrs_length, int[] out_values, int[] out_indices) { in RetrieveAttributes() argument
504 if (out_indices != null && in RetrieveAttributes()
[all …]
DAttributeResolution10.java96 int attrs_length, int[] out_values, int[] out_indices) { in ResolveAttrs() argument
227 if (out_indices != null && value.dataType != Res_value.TYPE_NULL) { in ResolveAttrs()
229 out_indices[indicesIdx] = ii; in ResolveAttrs()
235 if (out_indices != null) { in ResolveAttrs()
236 out_indices[0] = indicesIdx; in ResolveAttrs()
243 int[] out_values, int[] out_indices) { in ApplyStyle() argument
432 out_indices[indices_idx] = ii; in ApplyStyle()
451 out_indices[0] = indices_idx; in ApplyStyle()
455 int attrs_length, int[] out_values, int[] out_indices) { in RetrieveAttributes() argument
514 if (out_indices != null && in RetrieveAttributes()
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dsparsemask_op_test.py37 out_indices = np.array([2, 3, 4], dtype=np.int32)
51 self.assertAllEqual(tf_out_indices, out_indices)
Dlookup_ops_test.py199 out_indices, out_values, out_shape = self.evaluate(output)
202 self.assertAllEqual(sp_indices, out_indices)
/external/tensorflow/tensorflow/core/kernels/
Dcandidate_sampler_ops.cc237 Tensor* out_indices = nullptr; in Compute() local
241 0, TensorShape({static_cast<int>(indices.size())}), &out_indices)); in Compute()
253 out_indices->vec<int32>()(i) = indices[i]; in Compute()
Ddynamic_partition_op.cc151 Eigen::DSizes<Eigen::DenseIndex, 2> out_indices(oi, 0); in Compute() local
153 out_flat[p].slice(out_indices, sizes) = in Compute()
/external/robolectric-shadows/shadows/framework/src/main/java/org/robolectric/shadows/
DShadowArscAssetManager10.java1341 int[] out_indices = (int[])shadowVMRuntime.getObjectForAddress(out_indices_ptr); in nativeApplyStyle() local
1351 out_values, out_indices); in nativeApplyStyle()
1395 int[] out_indices = null; in nativeResolveAttrs() local
1399 out_indices = out_java_indices; in nativeResolveAttrs()
1401 if (out_indices == null) { in nativeResolveAttrs()
1420 attrs_len, out_values, out_indices); in nativeResolveAttrs()
1457 int[] out_indices = null; in nativeRetrieveAttributes() local
1461 out_indices = out_java_indices; in nativeRetrieveAttributes()
1463 if (out_indices == null) { in nativeRetrieveAttributes()
1477 out_indices); in nativeRetrieveAttributes()
DShadowArscAssetManager9.java1337 int[] out_indices = (int[])shadowVMRuntime.getObjectForAddress(out_indices_ptr); in nativeApplyStyle() local
1347 out_values, out_indices); in nativeApplyStyle()
1391 int[] out_indices = null; in nativeResolveAttrs() local
1395 out_indices = out_java_indices; in nativeResolveAttrs()
1397 if (out_indices == null) { in nativeResolveAttrs()
1416 attrs_len, out_values, out_indices); in nativeResolveAttrs()
1453 int[] out_indices = null; in nativeRetrieveAttributes() local
1457 out_indices = out_java_indices; in nativeRetrieveAttributes()
1459 if (out_indices == null) { in nativeRetrieveAttributes()
1473 out_indices); in nativeRetrieveAttributes()
/external/tensorflow/tensorflow/examples/speech_commands/
Dlabel_wav.cc74 Tensor* out_indices, Tensor* out_scores) { in GetTopLabels() argument
94 *out_indices = sorted_indices; in GetTopLabels()
/external/tensorflow/tensorflow/core/util/
Dexample_proto_fast_parsing.cc1494 Tensor* out_indices = &result->sparse_indices[d]; in FastParseSingleExample() local
1500 *out_indices = Tensor( in FastParseSingleExample()
1502 auto indices_flat = out_indices->flat<int64>(); in FastParseSingleExample()
2147 int64* out_indices = context_result->sparse_indices[t].flat<int64>().data(); in FastParseSequenceExample() local
2183 *out_indices++ = e; in FastParseSequenceExample()
2184 *out_indices++ = i; in FastParseSequenceExample()
2344 int64* out_indices = in FastParseSequenceExample() local
2393 *out_indices++ = e; in FastParseSequenceExample()
2394 *out_indices++ = num_rows; in FastParseSequenceExample()
2395 *out_indices++ = i; in FastParseSequenceExample()
/external/tensorflow/tensorflow/contrib/pi_examples/label_image/
Dlabel_image.cc231 Tensor* out_indices, Tensor* out_scores) { in GetTopLabels() argument
250 *out_indices = sorted_indices; in GetTopLabels()
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops.py2431 out_indices, out_values = gen_sparse_ops.sparse_sparse_maximum(
2439 return sparse_tensor.SparseTensor(out_indices, out_values, sp_a.dense_shape)
2469 out_indices, out_values = gen_sparse_ops.sparse_sparse_minimum(
2477 return sparse_tensor.SparseTensor(out_indices, out_values, sp_a.dense_shape)
Darray_ops.py1442 out_indices, to_gather = setdiff1d(indices, mask_indices)
1444 return ops.IndexedSlices(out_values, out_indices, a.dense_shape)
/external/tensorflow/tensorflow/contrib/pi_examples/camera/
Dcamera.cc259 Tensor* out_indices, Tensor* out_scores) { in GetTopLabels() argument
278 *out_indices = sorted_indices; in GetTopLabels()
/external/tensorflow/tensorflow/contrib/lookup/
Dlookup_ops_test.py179 out_indices, out_values, out_shape = sess.run(output)
182 self.assertAllEqual(sp_indices, out_indices)