Searched refs:in_strides (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tile_functor_gpu.h | 36 const int32* in_strides = buf; in TileKernel() local
43 i_idx += t / out_strides[i] % in_dim_sizes[i] * in_strides[i]; in TileKernel()
60 gtl::InlinedVector<int32, 8> in_strides = ComputeStride<int32>(in.shape()); in TileSimple() local
63 host_buf[i] = in_strides[i]; in TileSimple()
|
| D | tile_functor_cpu.h | 30 gtl::InlinedVector<int64, 8> in_strides = ComputeStride<int64>(in.shape()); in TileSimpleImpl() local
39 i_idx += t / out_strides[i] % in.dim_size(i) * in_strides[i]; in TileSimpleImpl()
|
| D | transpose_functor_cpu.cc | 38 gtl::InlinedVector<int64, 8> in_strides = ComputeStride<int64>(in.shape()); in TransposeSimple() local
42 auto transpose_fn = [=, &in_strides, &out_strides, &perm](int64 begin, in TransposeSimple()
50 i_idx += ratio * in_strides[perm[i]]; in TransposeSimple()
|
| D | transpose_functor_gpu.cu.cc | 38 const int32* in_strides = buf; in TransposeKernel() local
47 i_idx += ratio * in_strides[perm[i]]; in TransposeKernel()
66 gtl::InlinedVector<int32, 8> in_strides = ComputeStride<int32>(in.shape()); in TransposeSimple() local
70 host_buf[i] = in_strides[i]; in TransposeSimple()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/workgroups/ |
| D | ideal_workgroup_picker.cc | 40 HW in_strides) const { in ParamsAccepted()
42 strides == in_strides; in ParamsAccepted()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_transpose_op.cc | 134 memory::dims in_strides = CalculateTFStrides(in_dims); in MKLTransposeND() local
140 in.SetUsrMem(in_dims, in_strides, &in_tensor); in MKLTransposeND()
|