Home
last modified time | relevance | path

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

12

/external/mesa3d/src/compiler/nir/
Dnir_intrinsics.h326 #define SYSTEM_VALUE(name, components, num_indices, idx0, idx1, idx2) \ argument
427 #define LOAD(name, srcs, num_indices, idx0, idx1, idx2, flags) \ argument
462 #define STORE(name, srcs, num_indices, idx0, idx1, idx2, flags) \ argument
Dnir_serialize.c487 unsigned num_indices = nir_intrinsic_infos[intrin->intrinsic].num_indices; in write_intrinsic() local
513 unsigned num_indices = nir_intrinsic_infos[op].num_indices; in read_intrinsic() local
Dnir_builder.h628 num_variables, num_indices, idx0, idx1, idx2, flags) argument
Dnir.h945 num_variables, num_indices, idx0, idx1, idx2, flags) \ argument
1115 unsigned num_indices; member
/external/tensorflow/tensorflow/compiler/xla/
Dsparse_index_array.cc76 void SparseIndexArray::Resize(int64 num_indices) { in Resize()
85 int64 num_indices = index_count(); in Validate() 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.cc91 const int num_indices, in GetIndicesVector()
205 const int num_indices = SizeOfDimension(indices, 0); in SparseToDenseImpl() 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/tensorflow/tensorflow/core/kernels/
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
Dsegment_reduction_ops.cc59 const int64 num_indices = segment_ids.NumElements(); in SegmentReductionValidationHelper() local
90 const int64 num_indices = segment_ids.NumElements(); in Compute() local
228 const int64 num_indices = segment_ids.NumElements(); in ComputeAsync() local
653 const int64 num_indices = indices.NumElements(); in Compute() local
Dscatter_nd_op_gpu.cu.cc85 const Eigen::array<int64, IXDIM> batch_strides, const int64 num_indices, in ScatterNdOpKernel()
Dscatter_nd_op.cc56 bool ValidEmptyOutputShape(int64 num_inputs, int64 num_indices, in ValidEmptyOutputShape()
644 size_t num_indices = indices.NumElements(); in operator ()() local
Dtensor_array_ops.cc619 int32 num_indices; in Compute() 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.cc47 int64 num_indices = 1; in XlaGather() local
Dtensor_array_ops.cc376 const int num_indices = indices_shape.dim_size(0); in Compile() local
/external/v8/src/runtime/
Druntime-array.cc71 int num_indices = keys.is_null() ? limit : keys->length(); in RemoveArrayHolesGeneric() local
348 uint32_t num_indices = keys->length(); in CopyFromPrototype() local
/external/mesa3d/src/gallium/auxiliary/draw/
Ddraw_prim_assembler.c92 unsigned *indices, unsigned num_indices) in copy_verts()
/external/swiftshader/third_party/SPIRV-Tools/source/val/
Dvalidate_composites.cpp44 const uint32_t num_indices = num_words - word_index; in GetExtractInsertValueType() local
/external/deqp-deps/SPIRV-Tools/source/val/
Dvalidate_composites.cpp44 const uint32_t num_indices = num_words - word_index; in GetExtractInsertValueType() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dsegment_reduction_ops_test.py468 def _sparse_input(self, input_shape, num_indices, dtype=dtypes_lib.int32): argument
/external/mesa3d/src/compiler/spirv/
Dspirv_to_nir.c2847 unsigned num_indices) in vtn_composite_insert()
2873 const uint32_t *indices, unsigned num_indices) in vtn_composite_extract()
/external/flac/include/FLAC/
Dformat.h682 FLAC__byte num_indices; member
/external/flac/libFLAC/
Dmetadata_object.c343 static FLAC__StreamMetadata_CueSheet_Index *cuesheet_track_index_array_new_(unsigned num_indices) in cuesheet_track_index_array_new_()

12