| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_topk.cc | 29 tensorflow::int32* out_indices) { in TopK() argument
63 tensorflow::int32* out_indices_batch = out_indices + batch * k; in TopK()
74 tensorflow::int32* out_indices) { in __xla_cpu_runtime_TopKF32() argument
75 TopK(batch_size, input_size, k, values, out_values, out_indices); in __xla_cpu_runtime_TopKF32()
|
| D | runtime_topk.h | 29 tensorflow::int32* out_indices);
|
| /external/robolectric-shadows/resources/src/main/java/org/robolectric/res/android/ |
| D | AttributeResolution10.java | 96 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 …]
|
| D | AttributeResolution9.java | 92 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 …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparsemask_op_test.py | 37 out_indices = np.array([2, 3, 4], dtype=np.int32)
51 self.assertAllEqual(tf_out_indices, out_indices)
|
| D | lookup_ops_test.py | 228 out_indices, out_values, out_shape = self.evaluate(output)
231 self.assertAllEqual(sp_indices, out_indices)
|
| /external/tensorflow/tensorflow/core/util/ |
| D | bcast.h | 37 std::vector<int64>* out_indices) { in ComputeBatchIndices() argument
43 out_indices->resize(output_batch_size); in ComputeBatchIndices()
53 (*out_indices)[num_output_elements + k] = (*out_indices)[k] + incr; in ComputeBatchIndices()
|
| D | example_proto_fast_parsing.cc | 1720 Tensor* out_indices = &result->sparse_indices[d]; in FastParseSingleExample() local
1725 *out_indices = Tensor( in FastParseSingleExample()
1727 auto indices_flat = out_indices->flat<int64>(); in FastParseSingleExample()
2363 int64* out_indices = context_result->sparse_indices[t].flat<int64>().data(); in ParseContextSparseFeatures() local
2381 if (is_batch) *out_indices++ = e; in ParseContextSparseFeatures()
2382 *out_indices++ = i; in ParseContextSparseFeatures()
2669 int64* out_indices = in ParseSequenceSparseFeatures() local
2718 if (is_batch) *out_indices++ = e; in ParseSequenceSparseFeatures()
2719 *out_indices++ = num_rows; in ParseSequenceSparseFeatures()
2720 *out_indices++ = i; in ParseSequenceSparseFeatures()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | candidate_sampler_ops.cc | 237 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()
|
| D | dynamic_partition_op.cc | 151 Eigen::DSizes<Eigen::DenseIndex, 2> out_indices(oi, 0); in Compute() local
153 out_flat[p].slice(out_indices, sizes) = in Compute()
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| D | label_wav.cc | 75 Tensor* out_indices, Tensor* out_scores) { in GetTopLabels() argument
95 *out_indices = sorted_indices; in GetTopLabels()
|
| /external/robolectric-shadows/shadows/framework/src/main/java/org/robolectric/shadows/ |
| D | ShadowArscAssetManager9.java | 1337 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()
|
| D | ShadowArscAssetManager10.java | 1341 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()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_parallel_loops.cc | 265 SmallVector<Value, 1> out_indices; in CreateReduceOpInNestedParallelLoops() local
267 out_indices.reserve(outer.getNumLoops()); in CreateReduceOpInNestedParallelLoops()
269 out_indices.push_back(iv); in CreateReduceOpInNestedParallelLoops()
272 out_indices.push_back(rewriter->create<ConstantIndexOp>(loc, 0)); in CreateReduceOpInNestedParallelLoops()
275 rewriter->create<StoreOp>(loc, reduction_result, out, out_indices); in CreateReduceOpInNestedParallelLoops()
|
| /external/mesa3d/src/mesa/main/ |
| D | glthread_draw.c | 83 const GLvoid **out_indices) in upload_multi_indices() argument
102 out_indices[i] = (const GLvoid*)(intptr_t)(upload_offset + offset); in upload_multi_indices()
933 const GLvoid **out_indices = alloca(sizeof(indices[0]) * draw_count); in _mesa_marshal_MultiDrawElementsBaseVertex() local
937 out_indices); in _mesa_marshal_MultiDrawElementsBaseVertex()
938 indices = out_indices; in _mesa_marshal_MultiDrawElementsBaseVertex()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 2748 out_indices, out_values = gen_sparse_ops.sparse_sparse_maximum(
2756 return sparse_tensor.SparseTensor(out_indices, out_values, sp_a.dense_shape)
2792 out_indices, out_values = gen_sparse_ops.sparse_sparse_minimum(
2800 return sparse_tensor.SparseTensor(out_indices, out_values, sp_a.dense_shape)
|
| D | array_ops.py | 1964 out_indices, to_gather = gen_array_ops.list_diff(indices, mask_indices)
1966 return ops.IndexedSlices(out_values, out_indices, a.dense_shape)
|