| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| split_op.cc | 80 const int32 slice_size = input_shape.dim_size(split_dim) / num_split;
99 begin[split_dim] = i * slice_size;
100 limits[split_dim] = (i + 1) * slice_size;
159 int slice_size; variable
160 slice_size = split_size_literal.Get<int>({i});
161 if (slice_size == -1) {
169 split_sizes_vec[i] = slice_size;
170 total_split_size += slice_size;
200 int slice_size = split_sizes_vec[i]; variable
201 output_shape.set_dim(split_dim, slice_size);
[all...] |
| image_resize_ops.cc | 290 std::vector<int64> slice_size = in_size; variable
297 slice_size[i] = 1;
302 {batch, slice_size[0], slice_size[1], channels},
|
| /external/tensorflow/tensorflow/core/kernels/ |
| dynamic_partition_op.cc | 136 const int64 slice_size = data->NumElements() / N; variable
137 const auto data_flat = data->shaped<T, 2>({N, slice_size});
138 Eigen::DSizes<Eigen::DenseIndex, 2> sizes(1, slice_size);
|
| batch_util.cc | 161 Eigen::DSizes<Eigen::DenseIndex, NDIMS + 1> slice_size; local
162 slice_size[0] = 1;
163 for (size_t i = 1; i < slice_size.size(); ++i) {
164 slice_size[i] = element_t.dimension(i - 1);
166 parent_t.slice(slice_indices, slice_size) = element_t.reshape(slice_size);
|
| gather_functor.h | 117 const int64 slice_size = out.dimension(2); local
120 bool use_large = (slice_size > std::numeric_limits<int32>::max() ||
127 slice_size, out); \
129 const int32 small_slice = static_cast<int32>(slice_size); \
135 if (slice_size == 10)
137 else if (slice_size == 20)
|
| gather_functor_gpu.cu.h | 35 int64 slice_size, int64 out_size) {
41 indices_i = i / slice_size;
42 slice_i = i - indices_i * slice_size;
44 Index batch_indices_i = i / slice_size;
50 slice_i = i - batch_indices_i * slice_size;
62 // params is a [batch_size, gather_dim_size, slice_size] tensor. Read
66 (batch_i * gather_dim_size + gather_i) * slice_size + slice_i;
91 const int64 slice_size = params.dimension(2); local
99 indices_size, slice_size, out_size);
106 indices_size, slice_size, out_size)
[all...] |
| spectrogram_test.cc | 221 int slice_size = first_output[0].size(); local
223 ASSERT_EQ(slice_size, first_output[i].size());
224 ASSERT_EQ(slice_size, second_output[i].size());
225 for (int j = 0; j < slice_size; ++j) {
|
| gather_nd_op.cc | 141 const Index slice_size = static_cast<Index>(slice_size_big); local
160 auto out_mat = out->shaped<T, 2>({N_result, slice_size});
170 bad_i = func(c->eigen_device<Device>(), slice_size, scratch_scalar, \
210 const GPUDevice& d, const Index slice_size, \
|
| padding_fifo_queue.cc | 325 Eigen::DSizes<Eigen::DenseIndex, NDIMS + 1> slice_size; local
326 slice_size[0] = 1;
327 for (size_t i = 1; i < slice_size.size(); ++i) {
328 slice_size[i] = element_t.dimension(i - 1);
330 parent_t.slice(slice_indices, slice_size) = element_t.reshape(slice_size);
|
| dynamic_stitch_op.cc | 140 const int32 slice_size, const int32 first_dim_size,
175 const int slice_size = merged->flat_outer_dims<T>().dimension(1); variable
199 j * slice_size));
208 DynamicStitchGPUImpl<T>(c->eigen_gpu_device(), slice_size, first_dim_size,
240 const int slice_size = merged_flat.dimension(1); variable
241 const size_t slice_bytes = slice_size * sizeof(T);
247 data.shaped<T, 2>({indices_vec.dimension(0), slice_size});
257 memcpy(merged_base + index * slice_size, data_base + i * slice_size,
261 Eigen::DSizes<Eigen::DenseIndex, 2> sizes(1, slice_size);
[all...] |
| scatter_nd_op.cc | 333 Index* slice_size) {
393 *slice_size = static_cast<Index>(slice_size_big);
442 Index slice_size; local
444 shape, indices, updates, &slice_dim, &num_updates, &slice_size));
448 auto updates_flat = updates.shaped<T, 2>({num_updates, slice_size});
466 out->shaped<T, 2>({shape.num_elements() / slice_size, slice_size});
480 functor(c->eigen_device<Device>(), slice_size, output_shape_prefix, \
517 const GPUDevice& d, const Index slice_size, \
|
| /external/mesa3d/src/gallium/drivers/nouveau/nv50/ |
| nv98_video_bsp.c | 45 uint32_t slice_size, bucket_size, ring_size, bsp_size; local
136 nouveau_vp3_inter_sizes(dec, 1, &slice_size, &bucket_size, &ring_size);
140 PUSH_DATA (push, inter_addr + slice_size + bucket_size); // 408 interdata addr
148 nouveau_vp3_inter_sizes(dec, desc.h264->slice_count, &slice_size, &bucket_size, &ring_size);
152 PUSH_DATA (push, slice_size << 8); // 408 interparm size?
153 PUSH_DATA (push, inter_addr + slice_size + bucket_size); // 40c interdata addr
155 PUSH_DATA (push, inter_addr + slice_size); // 414 bucket?
|
| nv98_video_vp.c | 28 struct nouveau_bo *inter_bo, unsigned slice_size)
50 for (i = 0; i < comm->byte_ofs + slice_size; i += 0x10) {
75 uint32_t slice_size, bucket_size, ring_size, i; local
92 nouveau_vp3_inter_sizes(dec, desc.h264->slice_count, &slice_size, &bucket_size, &ring_size);
95 nouveau_vp3_inter_sizes(dec, 1, &slice_size, &bucket_size, &ring_size);
137 PUSH_DATA (push, inter_addr + slice_size + bucket_size); // 718 inter_data_ofs
144 PUSH_DATA (push, inter_addr + slice_size); // 720 bucket_ofs
187 dump_comm_vp(dec, dec->comm, comm_seq, inter_bo, slice_size << 8);
191 dump_comm_vp(dec, dec->comm, comm_seq, inter_bo, slice_size << 8);
|
| /external/mesa3d/src/gallium/drivers/nouveau/nvc0/ |
| nvc0_video_bsp.c | 142 uint32_t slice_size, bucket_size, ring_size; local
188 nouveau_vp3_inter_sizes(dec, 1, &slice_size, &bucket_size, &ring_size);
192 PUSH_DATA (push, inter_addr + slice_size + bucket_size); // 408 interdata addr
197 nouveau_vp3_inter_sizes(dec, desc.h264->slice_count, &slice_size, &bucket_size, &ring_size);
201 PUSH_DATA (push, slice_size << 8); // 408 interparm size?
202 PUSH_DATA (push, inter_addr + slice_size + bucket_size); // 40c interdata addr
204 PUSH_DATA (push, inter_addr + slice_size); // 414 bucket?
|
| nvc0_video_vp.c | 28 struct nouveau_bo *inter_bo, unsigned slice_size)
50 for (i = 0; i < comm->byte_ofs + slice_size; i += 0x10) {
75 uint32_t slice_size, bucket_size, ring_size, i; local
92 nouveau_vp3_inter_sizes(dec, desc.h264->slice_count, &slice_size, &bucket_size, &ring_size);
95 nouveau_vp3_inter_sizes(dec, 1, &slice_size, &bucket_size, &ring_size);
137 PUSH_DATA (push, inter_addr + slice_size + bucket_size); // 718 inter_data_ofs
144 PUSH_DATA (push, inter_addr + slice_size); // 720 bucket_ofs
187 dump_comm_vp(dec, dec->comm, comm_seq, inter_bo, slice_size << 8);
191 dump_comm_vp(dec, dec->comm, comm_seq, inter_bo, slice_size << 8);
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| split.cc | 59 const int slice_size = input_size / num_splits; local
63 output_dims->data[axis_value] = slice_size;
|
| /hardware/intel/common/libmix/mix_vbp/viddec_fw/fw/parser/ |
| vbp_vc1_parser.c | 930 uint32 slice_size = cxt->list.data[index].edpos - cxt->list.data[index].stpos; local
942 slc_data->slice_size = slice_size - byte;
945 slc_parms->slice_data_size = slc_data->slice_size;
|
| vbp_loader.h | 51 uint32 slice_size; member in struct:_vbp_slice_data_mp42
122 uint32 slice_size; /* slice data size */ member in struct:_vbp_slice_data_h264
213 uint32 slice_size; member in struct:_vbp_slice_data_vc1
|
| /hardware/qcom/media/msm8974/mm-video-legacy/vidc/common/src/ |
| extra_data_handler.cpp | 231 OMX_U32 slice_offset = 0, slice_size = 0, total_size = 0; local
256 slice_size = (OMX_U32)(*(data + (i*2 + 2)));
257 total_size += slice_size;
259 i, slice_offset, slice_size);
|
| /hardware/qcom/media/msm8974/mm-video-v4l2/vidc/common/src/ |
| extra_data_handler.cpp | 260 OMX_U32 slice_offset = 0, slice_size = 0, total_size = 0; local
290 slice_size = (OMX_U32)(*(data + (i*2 + 2)));
291 total_size += slice_size;
293 i, (unsigned int)slice_offset, (unsigned int)slice_size);
|
| /hardware/qcom/media/msm8996/mm-video-v4l2/vidc/common/src/ |
| extra_data_handler.cpp | 260 OMX_U32 slice_offset = 0, slice_size = 0, total_size = 0; local
290 slice_size = (OMX_U32)(*(data + (i*2 + 2)));
291 total_size += slice_size;
293 i, (unsigned int)slice_offset, (unsigned int)slice_size);
|
| /hardware/qcom/media/msm8998/mm-video-v4l2/vidc/common/src/ |
| extra_data_handler.cpp | 264 OMX_U32 slice_offset = 0, slice_size = 0, total_size = 0; local
294 slice_size = (OMX_U32)(*(data + (i*2 + 2)));
295 total_size += slice_size;
297 i, (unsigned int)slice_offset, (unsigned int)slice_size);
|
| /external/libdrm/radeon/ |
| radeon_surface.h | 71 uint64_t slice_size; member in struct:radeon_surface_level
|
| /external/mesa3d/src/gallium/drivers/ilo/core/ |
| ilo_image.c | 237 const uint64_t slice_size = info->width * info->height * local
241 const uint64_t estimated_size = slice_size * slice_count;
[all...] |
| /external/mesa3d/src/gallium/drivers/svga/include/ |
| svga3d_surfacedefs.h | 1023 uint32 slice_size, total_size; local
1038 slice_size = clamped_umul32(image_blocks.height, pitch);
1039 total_size = clamped_umul32(slice_size, image_blocks.depth);
[all...] |
