Home
last modified time | relevance | path

Searched defs:in_strides (Results 1 – 8 of 8) 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
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
Dtile_functor_cpu.h30 gtl::InlinedVector<int64_t, 8> in_strides = in TileSimpleImpl() local
Dtranspose_functor_cpu.cc38 gtl::InlinedVector<int64_t, 8> in_strides = in TransposeSimple() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_transpose_op.cc74 memory::dims in_strides = CalculateTFStrides(in_dims); in MKLTransposeND() local
/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/
Donedevice_train_palm_detection_filter_inference_beta.cc244 const int32_t in_strides[4] = { in DepthwiseConv2dNative() local
986 const int32_t in_strides[3] = { in TransposeRank3() local
1015 const int32_t in_strides[4] = { in TransposeRank4() local
Donedevice_train_palm_detection_filter_inference.cc245 const int32_t in_strides[4] = { in DepthwiseConv2dNative() local
987 const int32_t in_strides[3] = { in TransposeRank3() local
1016 const int32_t in_strides[4] = { in TransposeRank4() local
Donedevice_train_palm_detection_filter_inference_v2.cc244 const int32_t in_strides[4] = { in DepthwiseConv2dNative() local
986 const int32_t in_strides[3] = { in TransposeRank3() local
1015 const int32_t in_strides[4] = { in TransposeRank4() local