Searched defs:out_indices (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_topk.cc | 29 tensorflow::int32* out_indices) { in TopK()
74 tensorflow::int32* out_indices) { in __xla_cpu_runtime_TopKF32()
|
| /external/robolectric-shadows/resources/src/main/java/org/robolectric/res/android/ |
| D | AttributeResolution9.java | 92 int attrs_length, int[] out_values, int[] out_indices) { in ResolveAttrs()
239 int[] out_values, int[] out_indices) { in ApplyStyle()
445 int attrs_length, int[] out_values, int[] out_indices) { in RetrieveAttributes()
|
| D | AttributeResolution10.java | 96 int attrs_length, int[] out_values, int[] out_indices) { in ResolveAttrs()
243 int[] out_values, int[] out_indices) { in ApplyStyle()
455 int attrs_length, int[] out_values, int[] out_indices) { in RetrieveAttributes()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_partition_op.cc | 151 Eigen::DSizes<Eigen::DenseIndex, 2> out_indices(oi, 0); in Compute() local
|
| D | candidate_sampler_ops.cc | 237 Tensor* out_indices = nullptr; in Compute() local
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| D | label_wav.cc | 75 Tensor* out_indices, Tensor* out_scores) { in GetTopLabels()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | bcast.h | 37 std::vector<int64>* out_indices) { in ComputeBatchIndices()
|
| D | example_proto_fast_parsing.cc | 1720 Tensor* out_indices = &result->sparse_indices[d]; in FastParseSingleExample() local
2363 int64* out_indices = context_result->sparse_indices[t].flat<int64>().data(); in ParseContextSparseFeatures() local
2669 int64* out_indices = in ParseSequenceSparseFeatures() local
|
| /external/robolectric-shadows/shadows/framework/src/main/java/org/robolectric/shadows/ |
| D | ShadowArscAssetManager10.java | 1341 int[] out_indices = (int[])shadowVMRuntime.getObjectForAddress(out_indices_ptr); in nativeApplyStyle() local
1395 int[] out_indices = null; in nativeResolveAttrs() local
1457 int[] out_indices = null; in nativeRetrieveAttributes() local
|
| D | ShadowArscAssetManager9.java | 1337 int[] out_indices = (int[])shadowVMRuntime.getObjectForAddress(out_indices_ptr); in nativeApplyStyle() local
1391 int[] out_indices = null; in nativeResolveAttrs() local
1453 int[] out_indices = null; in nativeRetrieveAttributes() local
|
| /external/mesa3d/src/mesa/main/ |
| D | glthread_draw.c | 83 const GLvoid **out_indices) in upload_multi_indices()
933 const GLvoid **out_indices = alloca(sizeof(indices[0]) * draw_count); in _mesa_marshal_MultiDrawElementsBaseVertex() local
|
| /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
|