| /external/tensorflow/tensorflow/core/kernels/ |
| tensor_array_ops.cc | 646 if (output_shape.num_elements() == 0) {
653 output_tensor->shaped<T, 2>({1, output_shape.num_elements()});
[all...] |
| extract_image_patches_op.cc | 103 if (out_shape.num_elements() == 0) {
|
| histogram_op.cc | 86 OP_REQUIRES(ctx, (value_range_tensor.shape().num_elements() == 2),
|
| mkl_conv_ops.h | 386 if (input_tf_shape.num_elements() == 0 ||
387 filter_tf_shape.num_elements() == 0 ||
388 outbprop_tf_shape.num_elements() == 0) {
399 for (size_t i = 0; i < output_tf_shape.num_elements(); ++i) {
|
| ops_util.h | 56 const int64 bytes_per_dim0 = (s.num_elements() / dim0_size) * sizeof(T);
|
| reduce_join_op.cc | 164 for (int64 output_index = 0; output_index < output_shape.num_elements();
|
| scan_ops.cc | 67 if (output_shape.num_elements() == 0) return;
|
| stateless_random_ops.cc | 50 if (shape.num_elements() == 0) return;
|
| /external/tensorflow/tensorflow/core/util/ |
| example_proto_helper.h | 191 elements_per_stride.push_back(dense_shape.num_elements());
257 elements_per_stride.push_back(dense_shape.num_elements());
|
| /art/libartbase/base/ |
| hash_set.h | 436 // Reserve enough room to insert until Size() == num_elements without requiring to grow the hash
438 void Reserve(size_t num_elements) {
439 size_t num_buckets = num_elements / max_load_factor_;
441 while (static_cast<size_t>(num_buckets * max_load_factor_) <= num_elements + 1u) {
|
| /external/libxaac/decoder/ |
| ixheaacd_sbrdecoder.c | 262 WORD32 num_elements = p_sbr_bit_stream->no_elements; local
347 num_elements = p_sbr_bit_stream->no_elements;
349 num_elements = 0;
353 for (k = 0; k < num_elements; k++) {
|
| /external/mesa3d/src/compiler/nir/ |
| nir_clone.c | 134 nc->num_elements = c->num_elements;
135 nc->elements = ralloc_array(nvar, nir_constant *, c->num_elements);
136 for (unsigned i = 0; i < c->num_elements; i++) {
|
| /external/mesa3d/src/gallium/drivers/etnaviv/ |
| etnaviv_internal.h | 199 unsigned num_elements; member in struct:compiled_vertex_elements_state
|
| /external/mesa3d/src/gallium/drivers/radeonsi/ |
| si_debug.c | 368 unsigned num_elements,
381 for (i = 0; i < num_elements; i++) {
454 unsigned num_elements[] = { local
468 num_elements[i], f);
|
| /external/mesa3d/src/gallium/drivers/vc4/ |
| vc4_draw.c | 151 uint32_t num_elements_emit = MAX2(vtx->num_elements, 1);
185 for (int i = 0; i < vtx->num_elements; i++) {
211 if (vtx->num_elements == 0) {
226 assert(vtx->num_elements <= 8);
|
| vc4_state.c | 354 vc4_vertex_state_create(struct pipe_context *pctx, unsigned num_elements,
362 memcpy(so->pipe, elements, sizeof(*elements) * num_elements);
363 so->num_elements = num_elements;
|
| /external/mesa3d/src/gallium/drivers/virgl/ |
| virgl_encode.h | 135 unsigned num_elements,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| bcast_ops.cc | 100 for (int64 i = 0; i < in_shape.num_elements(); ++i) {
|
| scan_ops.cc | 71 if (input_shape.num_elements() == 0) {
|
| tile_ops.cc | 47 OP_REQUIRES(ctx, input_shape.dims() == multiples_shape.num_elements(),
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| parallel_loop_emitter.cc | 57 // if (linear_index < num_elements) {
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| device_setter.py | 136 return shape.num_elements() * elem_size
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| benchmark_model.cc | 203 int64 output_count = output_shape.num_elements();
216 int64 output_count = output_shape.num_elements();
223 int64 output_count = output_shape.num_elements();
605 int64 total_size = inputs[0].shape.num_elements();
|
| /external/webp/src/mux/ |
| muxread.c | 523 WebPChunkId id, int* num_elements) {
524 if (mux == NULL || num_elements == NULL) {
529 *num_elements = MuxImageCount(mux->images_, id);
533 *num_elements = CountChunks(*chunk_list, kChunks[idx].tag);
|
| /external/mesa3d/src/gallium/drivers/swr/ |
| swr_state.cpp | 449 unsigned num_elements,
453 assert(num_elements <= PIPE_MAX_ATTRIBS);
457 velems->fsState.numAttribs = num_elements;
458 for (unsigned i = 0; i < num_elements; i++) {
532 unsigned num_elements,
537 assert(num_elements <= PIPE_MAX_ATTRIBS);
543 num_elements);
[all...] |
