Searched refs:out_strides (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tile_functor_gpu.h | 37 const int32* out_strides = buf + ndims; in TileKernel() local
43 i_idx += t / out_strides[i] % in_dim_sizes[i] * in_strides[i]; in TileKernel()
44 t %= out_strides[i]; in TileKernel()
61 gtl::InlinedVector<int32, 8> out_strides = ComputeStride<int32>(out->shape()); in TileSimple() local
64 host_buf[ndims + i] = out_strides[i]; in TileSimple()
|
| D | tile_functor_cpu.h | 31 gtl::InlinedVector<int64, 8> out_strides = ComputeStride<int64>(out->shape()); in TileSimpleImpl() local
39 i_idx += t / out_strides[i] % in.dim_size(i) * in_strides[i]; in TileSimpleImpl()
40 t %= out_strides[i]; in TileSimpleImpl()
|
| D | transpose_functor_cpu.cc | 39 gtl::InlinedVector<int64, 8> out_strides = ComputeStride<int64>(out->shape()); in TransposeSimple() local
42 auto transpose_fn = [=, &in_strides, &out_strides, &perm](int64 begin, in TransposeSimple()
48 const int64 ratio = t / out_strides[i]; in TransposeSimple()
49 t -= ratio * out_strides[i]; in TransposeSimple()
|
| D | transpose_functor_gpu.cu.cc | 39 const int32* out_strides = buf + ndims; in TransposeKernel() local
45 const int32 ratio = t / out_strides[i]; in TransposeKernel()
46 t -= ratio * out_strides[i]; in TransposeKernel()
67 gtl::InlinedVector<int32, 8> out_strides = ComputeStride<int32>(out->shape()); in TransposeSimple() local
71 host_buf[ndims + i] = out_strides[i]; in TransposeSimple()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_batch_matmul_op.cc | 205 auto out_strides = CalculateTFStrides(out_dims); in CreateMatMulParams() local
227 lhs_dims, rhs_dims, out_dims, lhs_strides, rhs_strides, out_strides); in CreateMatMulParams()
|
| D | mkl_transpose_op.cc | 136 memory::dims out_strides = in MKLTransposeND() local
143 out.SetUsrMem(in_dims, out_strides, out_tensor); in MKLTransposeND()
|
| /external/virglrenderer/src/ |
| D | vrend_winsys_gbm.c | 468 query->out_strides[plane] = 0; in virgl_gbm_export_query()
474 query->out_strides[plane] = gbm_bo_get_stride_for_plane(bo, plane); in virgl_gbm_export_query()
505 query->out_strides[plane] = 0; in virgl_gbm_export_query()
|
| D | virglrenderer.h | 165 uint32_t out_strides[4]; member
|
| D | virglrenderer.c | 719 export_query->out_strides[0] = 0; in virgl_renderer_export_query()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | ceva_tflm_lib.h | 283 float* out_data, const int32_t* out_strides,
289 const int32_t* out_strides, const int32_t* dims,
|
| /external/crosvm/rutabaga_gfx/src/generated/ |
| D | virgl_renderer_bindings.rs | 170 pub out_strides: [u32; 4usize], field
|
| /external/crosvm/rutabaga_gfx/src/ |
| D | virgl_renderer.rs | 246 strides: query.out_strides, in query()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1387 SmallVector<int64_t, 4> out_strides; in GetOrCreateArrayView() local
1388 if (failed(getStridesAndOffset(out_memref_type, out_strides, out_offset))) in GetOrCreateArrayView()
1393 out_strides); in GetOrCreateArrayView()
|