Searched defs:out_strides (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tile_functor_gpu.h | 37 const int32* out_strides = buf + ndims; in TileKernel() local
61 gtl::InlinedVector<int32, 8> out_strides = ComputeStride<int32>(out->shape()); in TileSimple() local
|
| D | transpose_functor_gpu.cu.cc | 39 const int32* out_strides = buf + ndims; in TransposeKernel() local
67 gtl::InlinedVector<int32, 8> out_strides = ComputeStride<int32>(out->shape()); in TransposeSimple() local
|
| D | tile_functor_cpu.h | 32 gtl::InlinedVector<int64_t, 8> out_strides = in TileSimpleImpl() local
|
| D | transpose_functor_cpu.cc | 40 gtl::InlinedVector<int64_t, 8> out_strides = in TransposeSimple() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_batch_matmul_helper.h | 79 auto out_strides = CalculateTFStrides(out_dims); in CreateMatMulParams() local
|
| D | mkl_transpose_op.cc | 76 memory::dims out_strides = in MKLTransposeND() local
|
| /external/virglrenderer/src/ |
| D | virglrenderer.h | 219 uint32_t out_strides[4]; member
|
| /external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| D | onedevice_train_palm_detection_filter_inference_beta.cc | 991 const int32_t out_strides[3] = {in_strides[perm[0]], in_strides[perm[1]], in TransposeRank3() local
1021 const int32_t out_strides[4] = {in_strides[perm[0]], in_strides[perm[1]], in TransposeRank4() local
|
| D | onedevice_train_palm_detection_filter_inference.cc | 992 const int32_t out_strides[3] = {in_strides[perm[0]], in_strides[perm[1]], in TransposeRank3() local
1022 const int32_t out_strides[4] = {in_strides[perm[0]], in_strides[perm[1]], in TransposeRank4() local
|
| D | onedevice_train_palm_detection_filter_inference_v2.cc | 991 const int32_t out_strides[3] = {in_strides[perm[0]], in_strides[perm[1]], in TransposeRank3() local
1021 const int32_t out_strides[4] = {in_strides[perm[0]], in_strides[perm[1]], in TransposeRank4() local
|
| /external/crosvm/rutabaga_gfx/src/generated/ |
| D | virgl_renderer_bindings.rs | 188 pub out_strides: [u32; 4usize], field
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1477 SmallVector<int64_t, 4> out_strides; in GetOrCreateArrayView() local
|