| /external/tensorflow/tensorflow/core/kernels/ |
| tile_functor_cpu.cc | 31 gtl::InlinedVector<int64, 8> in_strides = ComputeStride<int64>(in.shape()); local
40 i_idx += t / out_strides[i] % in.dim_size(i) * in_strides[i];
|
| mkl_transpose_op.cc | 133 memory::dims in_strides = CalculateTFStrides(in_dims); local
138 in.SetUsrMem(in_dims, in_strides, &in_tensor);
|
| tile_functor_gpu.h | 36 const int32* in_strides = buf; local
43 i_idx += t / out_strides[i] % in_dim_sizes[i] * in_strides[i];
60 gtl::InlinedVector<int32, 8> in_strides = ComputeStride<int32>(in.shape()); local
63 host_buf[i] = in_strides[i];
|
| transpose_functor_cpu.cc | 38 gtl::InlinedVector<int64, 8> in_strides = ComputeStride<int64>(in.shape()); local
42 auto transpose_fn = [=, &in_strides, &out_strides, &perm](int64 begin,
50 i_idx += ratio * in_strides[perm[i]];
|
| transpose_functor_gpu.cu.cc | 37 const int32* in_strides = buf; local
46 i_idx += ratio * in_strides[perm[i]];
65 gtl::InlinedVector<int32, 8> in_strides = ComputeStride<int32>(in.shape()); local
69 host_buf[i] = in_strides[i];
|
| mkl_conv_grad_bias_ops.cc | 113 GetStridesFromSizes(data_format_, mkl_context.in_strides,
151 size_t in_strides[4]; member in struct:tensorflow::MklConv2DCustomBackpropBiasOp::__anon45424
175 CHECK_EQ(dnnLayoutCreate_F32(<_input, in_dims, in_sizes, in_strides),
|
| mkl_pooling_ops_common.h | 761 size_t in_strides[4]; member in struct:tensorflow::__anon45429
|
| mkl_conv_ops.cc | 582 GetStridesFromSizes(data_format_, mkl_context.in_strides,
688 size_t in_strides[4]; member in struct:tensorflow::MklConvOp::__anon45425
710 CHECK_EQ(dnnLayoutCreate_F32(<_input, in_dims, in_sizes, in_strides),
[all...] |
