| /external/tensorflow/tensorflow/core/kernels/ |
| tile_functor_cpu.cc | 32 gtl::InlinedVector<int64, 8> out_strides = ComputeStride<int64>(out->shape()); local
40 i_idx += t / out_strides[i] % in.dim_size(i) * in_strides[i];
41 t %= out_strides[i];
|
| mkl_transpose_op.cc | 135 memory::dims out_strides = local
141 out.SetUsrMem(in_dims, out_strides, out_tensor);
|
| tile_functor_gpu.h | 37 const int32* out_strides = buf + ndims; local
43 i_idx += t / out_strides[i] % in_dim_sizes[i] * in_strides[i];
44 t %= out_strides[i];
61 gtl::InlinedVector<int32, 8> out_strides = ComputeStride<int32>(out->shape()); local
64 host_buf[ndims + i] = out_strides[i];
|
| transpose_functor_cpu.cc | 39 gtl::InlinedVector<int64, 8> out_strides = ComputeStride<int64>(out->shape()); local
42 auto transpose_fn = [=, &in_strides, &out_strides, &perm](int64 begin,
48 const int64 ratio = t / out_strides[i];
49 t -= ratio * out_strides[i];
|
| transpose_functor_gpu.cu.cc | 38 const int32* out_strides = buf + ndims; local
44 const int32 ratio = t / out_strides[i];
45 t -= ratio * out_strides[i];
66 gtl::InlinedVector<int32, 8> out_strides = ComputeStride<int32>(out->shape()); local
70 host_buf[ndims + i] = out_strides[i];
|
| mkl_conv_grad_bias_ops.cc | 118 mkl_context.out_strides[0] = 1;
153 size_t out_strides[1]; member in struct:tensorflow::MklConv2DCustomBackpropBiasOp::__anon45424
170 CHECK_EQ(dnnLayoutCreate_F32(<_outbackprop, 1, out_sizes, out_strides),
|
| mkl_pooling_ops_common.h | 763 size_t out_strides[4]; member in struct:tensorflow::__anon45429
|
| mkl_conv_ops.cc | 580 GetStridesFromSizes(data_format_, mkl_context.out_strides,
630 mkl_context.out_strides);
690 size_t out_strides[4]; member in struct:tensorflow::MklConvOp::__anon45425
[all...] |
