Home
last modified time | relevance | path

Searched defs:num_indices (Results 1 – 25 of 58) sorted by relevance

123

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DLegacyThrustHelpers.cu22 …st_helper(Tensor &linearIndex, Tensor &orig_indices, Tensor &sorted_indices, int64_t num_indices) { in index_put_with_sort_kernel_thrust_helper()
52 auto num_indices = count.numel(); in embedding_dense_backward_cuda_scan() local
H A DEmbedding.cu258 auto num_indices = indices.numel(); in embedding_dense_backward_cuda() local
358 auto num_indices = indices.numel(); in embedding_renorm_cuda_() local
/aosp_15_r20/external/libaom/aom_dsp/flow_estimation/
H A Dransac.c141 int num_indices, double *params) { in find_translation()
167 int num_indices, double *params) { in find_rotzoom()
210 int num_indices, double *params) { in find_affine()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsegment_reduction_ops_impl_1.cc30 const int64_t num_indices = segment_ids.NumElements(); in ValidateSegmentReduction() local
90 const int64_t num_indices = indices.NumElements(); in ValidateSparseSegmentReduction() local
H A Dsegment_reduction_ops_impl.h97 const int64_t num_indices = segment_ids.NumElements(); in Compute() local
242 const int64_t num_indices = segment_ids.NumElements(); in ComputeAsync() local
H A Dstring_split_op.cc200 std::vector<int64_t> num_indices(batch_size); in Compute() local
273 std::vector<int64_t> num_indices(batch_size); in Compute() local
H A Dctc_loss_op.cc58 int num_indices, int batch_size, in DoHistogram()
312 auto num_indices = labels_indices->dim_size(0); in Compute() local
H A Dresource_variable_ops.cc966 const Tensor& updates, Index num_indices) { in DoScatterOnCpu()
1000 const Tensor& updates, Index num_indices) { in DoScatter()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dgather.cc125 const size_t num_indices = positions->bytes / sizeof(PositionsT); in Gather() local
152 const size_t num_indices = positions->bytes / sizeof(PositionT); in GatherStrings() local
H A Dsparse_to_dense.cc87 const int num_indices, in GetIndicesVector()
223 const int num_indices = SizeOfDimension(indices, 0); in SparseToDenseImpl() local
H A Dgather_nd.cc127 const size_t num_indices = indices->bytes / sizeof(IndicesT); in EvalGatherNd() local
/aosp_15_r20/external/flac/oss-fuzz/
H A Dmetadata.cc284 uint32_t num_tracks, num_indices; in run_tests_with_level_2_interface() local
371 uint32_t num_tracks, num_indices; in run_tests_with_level_2_interface() local
/aosp_15_r20/external/mesa3d/src/freedreno/decode/
H A Dcffdec.c1081 __do_query(const char *primtype, uint32_t num_indices) in __do_query()
1133 do_query_compare(const char *primtype, uint32_t num_indices) in do_query_compare()
1179 do_query(const char *primtype, uint32_t num_indices) in do_query()
1964 uint32_t num_indices = dwords[2]; in draw_indx_common() local
1995 uint32_t num_indices = draw_indx_common(dwords, level); in cp_draw_indx() local
2039 uint32_t num_indices = draw_indx_common(dwords, level); in cp_draw_indx_2() local
2079 uint32_t num_indices = dwords[2]; in cp_draw_indx_offset() local
/aosp_15_r20/external/mesa3d/src/compiler/spirv/
H A Dvtn_cmat.c266 const uint32_t *indices, unsigned num_indices) in vtn_cooperative_matrix_extract()
283 unsigned num_indices) in vtn_cooperative_matrix_insert()
/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorRef.h211 const std::size_t num_indices = (sizeof...(otherIndices) + 1); in operator() local
218 const std::size_t num_indices = (sizeof...(otherIndices) + 1); in coeffRef() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Drandom_index_shuffle_ops.cc74 const int64_t num_indices = index_rank ? c->Value(c->Dim(index_shape, 0)) : 1; in StatelessRandomPermuteShape() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_index_splitter.cc50 int64_t num_indices = dynamic_op->operand(0)->shape().rank(); in Run() local
/aosp_15_r20/external/libgav1/src/
H A Dscan_test.cc47 int num_indices; in TEST_P() local
/aosp_15_r20/external/mesa3d/src/gallium/auxiliary/tessellator/
H A Dp_tessellator.h44 uint32_t num_indices; member
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dpywrap_gradient_exclusions.cc33 int num_indices; member
/aosp_15_r20/external/icing/icing/tokenization/reverse_jni/
H A Dreverse-jni-break-iterator.cc164 jint num_indices = jni_cache_->GetEnv()->GetArrayLength(break_indices.get()); in FetchNextBatch() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/fuzzing/
H A Dscatter_nd_fuzz.cc83 size_t num_indices = 1; in FuzzImpl() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/lib/
H A Dscatter.cc61 int64_t num_indices = 1; in XlaScatter() local
/aosp_15_r20/external/mesa3d/src/gallium/frontends/d3d10umd/
H A DDraw.cpp115 create_null_index_buffer(struct pipe_context *ctx, uint num_indices, in create_null_index_buffer()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dgather_op.cc54 int64_t num_indices = 1; in XlaGather() local

123