Home
last modified time | relevance | path

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

12

/external/rust/crates/vulkano/src/command_buffer/validity/
Dindex_buffer.rs56 pub num_indices: u32, field
101 fn num_indices() { in num_indices() function
/external/tensorflow/tensorflow/core/kernels/
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
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
Dctc_loss_op.cc58 int num_indices, int batch_size, in DoHistogram()
312 auto num_indices = labels_indices->dim_size(0); in Compute() local
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
Dscatter_nd_op_gpu.cu.cc101 const Eigen::array<int64, IXDIM> batch_strides, const int64 num_indices, in ScatterNdOpKernel()
/external/tensorflow/tensorflow/lite/kernels/
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
Dsparse_to_dense.cc87 const int num_indices, in GetIndicesVector()
223 const int num_indices = SizeOfDimension(indices, 0); in SparseToDenseImpl() local
Dgather_nd.cc127 const size_t num_indices = indices->bytes / sizeof(IndicesT); in EvalGatherNd() local
/external/tensorflow/tensorflow/core/ops/
Drandom_index_shuffle_ops.cc74 const int64_t num_indices = index_rank ? c->Value(c->Dim(index_shape, 0)) : 1; in StatelessRandomPermuteShape() local
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_index_splitter.cc50 int64_t num_indices = dynamic_op->operand(0)->shape().rank(); in Run() local
/external/libgav1/src/
Dscan_test.cc47 int num_indices; in TEST_P() local
/external/mesa3d/src/gallium/auxiliary/tessellator/
Dp_tessellator.h44 uint32_t num_indices; member
/external/flac/oss-fuzz/
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
/external/tensorflow/tensorflow/python/eager/
Dpywrap_gradient_exclusions.cc33 int num_indices; member
/external/icing/icing/tokenization/reverse_jni/
Dreverse-jni-break-iterator.cc164 jint num_indices = jni_cache_->GetEnv()->GetArrayLength(break_indices.get()); in FetchNextBatch() local
/external/tensorflow/tensorflow/core/kernels/fuzzing/
Dscatter_nd_fuzz.cc83 size_t num_indices = 1; in FuzzImpl() local
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dscatter.cc61 int64_t num_indices = 1; in XlaScatter() local
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
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
/external/mesa3d/src/freedreno/decode/
Dcffdec.c1008 __do_query(const char *primtype, uint32_t num_indices) in __do_query()
1051 do_query_compare(const char *primtype, uint32_t num_indices) in do_query_compare()
1097 do_query(const char *primtype, uint32_t num_indices) in do_query()
1853 uint32_t num_indices = dwords[2]; in draw_indx_common() local
1886 uint32_t num_indices = draw_indx_common(dwords, level); in cp_draw_indx() local
1930 uint32_t num_indices = draw_indx_common(dwords, level); in cp_draw_indx_2() local
1970 uint32_t num_indices = dwords[2]; in cp_draw_indx_offset() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dgather_op.cc54 int64_t num_indices = 1; in XlaGather() local
/external/mesa3d/src/gallium/auxiliary/draw/
Ddraw_prim_assembler.c101 unsigned *indices, unsigned num_indices) in copy_verts()
/external/mesa3d/src/gallium/drivers/radeonsi/
Dsi_compute_prim_discard.c693 LLVMValueRef num_indices = LLVMBuildMul( in si_build_prim_discard_compute_shader() local
699 LLVMValueRef num_indices = LLVMBuildMul( in si_build_prim_discard_compute_shader() local
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/val/
Dvalidate_composites.cpp42 const uint32_t num_indices = num_words - word_index; in GetExtractInsertValueType() local
/external/swiftshader/third_party/SPIRV-Tools/source/val/
Dvalidate_composites.cpp44 const uint32_t num_indices = num_words - word_index; in GetExtractInsertValueType() local

12