Home
last modified time | relevance | path

Searched defs:in_strides (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dtile_functor_gpu.h36 const int32* in_strides = buf; in TileKernel() local
60 gtl::InlinedVector<int32, 8> in_strides = ComputeStride<int32>(in.shape()); in TileSimple() local
Dtile_functor_cpu.h30 gtl::InlinedVector<int64, 8> in_strides = ComputeStride<int64>(in.shape()); in TileSimpleImpl() local
Dtranspose_functor_gpu.cu.cc38 const int32* in_strides = buf; in TransposeKernel() local
66 gtl::InlinedVector<int32, 8> in_strides = ComputeStride<int32>(in.shape()); in TransposeSimple() local
Dtranspose_functor_cpu.cc38 gtl::InlinedVector<int64, 8> in_strides = ComputeStride<int64>(in.shape()); in TransposeSimple() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_transpose_op.cc134 memory::dims in_strides = CalculateTFStrides(in_dims); in MKLTransposeND() local