Home
last modified time | relevance | path

Searched refs:out_strides (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dtile_functor_gpu.h37 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()
Dtile_functor_cpu.cc32 gtl::InlinedVector<int64, 8> out_strides = ComputeStride<int64>(out->shape()); in TileSimple() local
40 i_idx += t / out_strides[i] % in.dim_size(i) * in_strides[i]; in TileSimple()
41 t %= out_strides[i]; in TileSimple()
Dtranspose_functor_gpu.cu.cc38 const int32* out_strides = buf + ndims; in TransposeKernel() local
44 const int32 ratio = t / out_strides[i]; in TransposeKernel()
45 t -= ratio * out_strides[i]; in TransposeKernel()
66 gtl::InlinedVector<int32, 8> out_strides = ComputeStride<int32>(out->shape()); in TransposeSimple() local
70 host_buf[ndims + i] = out_strides[i]; in TransposeSimple()
Dtranspose_functor_cpu.cc39 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()
Dmkl_conv_grad_bias_ops.cc118 mkl_context.out_strides[0] = 1; in Compute()
153 size_t out_strides[1]; member
170 CHECK_EQ(dnnLayoutCreate_F32(&lt_outbackprop, 1, out_sizes, out_strides), in MklCreateInputLayouts()
Dmkl_transpose_op.cc135 memory::dims out_strides = in MKLTransposeND() local
141 out.SetUsrMem(in_dims, out_strides, out_tensor); in MKLTransposeND()
Dmkl_avgpooling_op.cc130 mkl_context.params.out_strides); in Compute()
382 params.out_sizes, params.out_strides), in MklCreateLayoutsAndPrimitives()
Dmkl_maxpooling_op.cc111 mkl_context.params.out_strides); in Compute()
335 params.out_sizes, params.out_strides), in MklCreateLayouts()
Dmkl_pooling_ops_common.cc421 GetStridesFromSizes(data_format, mkl_params->out_strides, in ExtractMklOpParams()
Dmkl_conv_ops.cc580 GetStridesFromSizes(data_format_, mkl_context.out_strides, in Compute()
630 mkl_context.out_strides); in Compute()
690 size_t out_strides[4]; member
Dmkl_pooling_ops_common.h763 size_t out_strides[4]; member