Searched refs:in_strides (Results 1 – 11 of 11) 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 | transpose_functor_gpu.cu.cc | 37 const int32* in_strides = buf; in TransposeKernel() local
46 i_idx += ratio * in_strides[perm[i]]; in TransposeKernel()
65 gtl::InlinedVector<int32, 8> in_strides = ComputeStride<int32>(in.shape()); in TransposeSimple() local
69 host_buf[i] = in_strides[i]; in TransposeSimple()
|
| D | tile_functor_cpu.cc | 31 gtl::InlinedVector<int64, 8> in_strides = ComputeStride<int64>(in.shape()); in TileSimple() local
40 i_idx += t / out_strides[i] % in.dim_size(i) * in_strides[i]; in TileSimple()
|
| 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 | mkl_conv_grad_bias_ops.cc | 113 GetStridesFromSizes(data_format_, mkl_context.in_strides, in Compute()
151 size_t in_strides[4]; member
175 CHECK_EQ(dnnLayoutCreate_F32(<_input, in_dims, in_sizes, in_strides), in MklCreateInputLayouts()
|
| D | mkl_transpose_op.cc | 133 memory::dims in_strides = CalculateTFStrides(in_dims); in MKLTransposeND() local
138 in.SetUsrMem(in_dims, in_strides, &in_tensor); in MKLTransposeND()
|
| D | mkl_avgpooling_op.cc | 167 params.in_sizes, params.in_strides), in MklCreateLayoutsAndPrimitives()
321 mkl_context.params.in_strides); in Compute()
391 params.in_sizes, params.in_strides), in MklCreateLayoutsAndPrimitives()
|
| D | mkl_maxpooling_op.cc | 159 params.in_sizes, params.in_strides), in MklCreateLayoutsAndPrimitives()
284 mkl_context.params.in_strides); in Compute()
325 params.in_sizes, params.in_strides), in MklCreateLayouts()
|
| D | mkl_pooling_ops_common.cc | 413 GetStridesFromSizes(data_format, mkl_params->in_strides, in ExtractMklOpParams()
|
| D | mkl_conv_ops.cc | 582 GetStridesFromSizes(data_format_, mkl_context.in_strides, in Compute()
688 size_t in_strides[4]; member
710 CHECK_EQ(dnnLayoutCreate_F32(<_input, in_dims, in_sizes, in_strides), in MklCreateInputLayouts()
|
| D | mkl_pooling_ops_common.h | 761 size_t in_strides[4]; member
|