Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
Dsegment_reduction_ops_impl.h99 const int64 num_indices = segment_ids.NumElements(); in Compute() local
239 const int64 num_indices = segment_ids.NumElements(); in ComputeAsync() local
Dstring_split_op.cc200 std::vector<int64> num_indices(batch_size); in Compute() local
273 std::vector<int64> num_indices(batch_size); in Compute() local
Dctc_loss_op.cc56 int num_indices, int batch_size, in DoHistogram()
303 auto num_indices = labels_indices->dim_size(0); in Compute() local
Dsegment_reduction_ops_impl_1.cc29 const int64 num_indices = segment_ids.NumElements(); in SegmentReductionValidationHelper() local
Dscatter_nd_op_gpu.cu.cc101 const Eigen::array<int64, IXDIM> batch_strides, const int64 num_indices, in ScatterNdOpKernel()
Dtensor_array_ops.cc614 int32 num_indices; in Compute() local
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_index_splitter.cc48 int64 num_indices = dynamic_op->operand(0)->shape().rank(); in Run() local
/external/tensorflow/tensorflow/lite/kernels/
Dsparse_to_dense.cc87 const int num_indices, in GetIndicesVector()
223 const int num_indices = SizeOfDimension(indices, 0); in SparseToDenseImpl() local
/external/mesa3d/src/gallium/auxiliary/tessellator/
Dp_tessellator.h44 uint32_t num_indices; member
/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 num_indices = 1; in XlaScatter() local
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorRef.h200 const std::size_t num_indices = (sizeof...(otherIndices) + 1); in operator() local
207 const std::size_t num_indices = (sizeof...(otherIndices) + 1); in coeffRef() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dgather_op.cc53 int64 num_indices = 1; in XlaGather() local
Dtensor_array_ops.cc376 const int num_indices = indices_shape.dim_size(0); in Compile() 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/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/deqp-deps/SPIRV-Tools/source/val/
Dvalidate_composites.cpp43 const uint32_t num_indices = num_words - word_index; in GetExtractInsertValueType() local
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/val/
Dvalidate_composites.cpp43 const uint32_t num_indices = num_words - word_index; in GetExtractInsertValueType() local
/external/swiftshader/third_party/SPIRV-Tools/source/val/
Dvalidate_composites.cpp43 const uint32_t num_indices = num_words - word_index; in GetExtractInsertValueType() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dsegment_reduction_ops_test.py506 def _sparse_input(self, input_shape, num_indices, dtype=dtypes_lib.int32): argument
/external/icing/icing/
Dicing-search-engine_benchmark.cc358 int num_indices = state.range(0); in BM_MutlipleIndices() local
/external/llvm-project/lldb/source/Plugins/ExpressionParser/Clang/
DClangExpressionDeclMap.cpp1125 uint32_t num_indices = sc_list.GetSize(); in SearchFunctionsInSymbolContexts() local
1270 for (uint32_t index = 0, num_indices = sc_list.GetSize(); in LookupFunction() local

12