/external/tensorflow/tensorflow/core/kernels/ |
D | segment_reduction_ops_impl.h | 99 const int64 num_indices = segment_ids.NumElements(); in Compute() local 239 const int64 num_indices = segment_ids.NumElements(); in ComputeAsync() local
|
D | string_split_op.cc | 200 std::vector<int64> num_indices(batch_size); in Compute() local 273 std::vector<int64> num_indices(batch_size); in Compute() local
|
D | ctc_loss_op.cc | 56 int num_indices, int batch_size, in DoHistogram() 303 auto num_indices = labels_indices->dim_size(0); in Compute() local
|
D | segment_reduction_ops_impl_1.cc | 29 const int64 num_indices = segment_ids.NumElements(); in SegmentReductionValidationHelper() local
|
D | scatter_nd_op_gpu.cu.cc | 101 const Eigen::array<int64, IXDIM> batch_strides, const int64 num_indices, in ScatterNdOpKernel()
|
D | tensor_array_ops.cc | 614 int32 num_indices; in Compute() local
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | dynamic_index_splitter.cc | 48 int64 num_indices = dynamic_op->operand(0)->shape().rank(); in Run() local
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | sparse_to_dense.cc | 87 const int num_indices, in GetIndicesVector() 223 const int num_indices = SizeOfDimension(indices, 0); in SparseToDenseImpl() local
|
/external/mesa3d/src/gallium/auxiliary/tessellator/ |
D | p_tessellator.h | 44 uint32_t num_indices; member
|
/external/tensorflow/tensorflow/python/eager/ |
D | pywrap_gradient_exclusions.cc | 33 int num_indices; member
|
/external/icing/icing/tokenization/reverse_jni/ |
D | reverse-jni-break-iterator.cc | 164 jint num_indices = jni_cache_->GetEnv()->GetArrayLength(break_indices.get()); in FetchNextBatch() local
|
/external/tensorflow/tensorflow/core/kernels/fuzzing/ |
D | scatter_nd_fuzz.cc | 83 size_t num_indices = 1; in FuzzImpl() local
|
/external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
D | scatter.cc | 61 int64 num_indices = 1; in XlaScatter() local
|
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
D | TensorRef.h | 200 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/ |
D | gather_op.cc | 53 int64 num_indices = 1; in XlaGather() local
|
D | tensor_array_ops.cc | 376 const int num_indices = indices_shape.dim_size(0); in Compile() local
|
/external/mesa3d/src/freedreno/decode/ |
D | cffdec.c | 1008 __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/ |
D | draw_prim_assembler.c | 101 unsigned *indices, unsigned num_indices) in copy_verts()
|
/external/mesa3d/src/gallium/drivers/radeonsi/ |
D | si_compute_prim_discard.c | 693 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/ |
D | validate_composites.cpp | 43 const uint32_t num_indices = num_words - word_index; in GetExtractInsertValueType() local
|
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/val/ |
D | validate_composites.cpp | 43 const uint32_t num_indices = num_words - word_index; in GetExtractInsertValueType() local
|
/external/swiftshader/third_party/SPIRV-Tools/source/val/ |
D | validate_composites.cpp | 43 const uint32_t num_indices = num_words - word_index; in GetExtractInsertValueType() local
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | segment_reduction_ops_test.py | 506 def _sparse_input(self, input_shape, num_indices, dtype=dtypes_lib.int32): argument
|
/external/icing/icing/ |
D | icing-search-engine_benchmark.cc | 358 int num_indices = state.range(0); in BM_MutlipleIndices() local
|
/external/llvm-project/lldb/source/Plugins/ExpressionParser/Clang/ |
D | ClangExpressionDeclMap.cpp | 1125 uint32_t num_indices = sc_list.GetSize(); in SearchFunctionsInSymbolContexts() local 1270 for (uint32_t index = 0, num_indices = sc_list.GetSize(); in LookupFunction() local
|