Searched defs:inner_dim (Results 1 - 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | cross_op.cc | 38 auto inner_dim = in0_shape.dim_size(in0_shape.dims() - 1); variable 39 OP_REQUIRES(ctx, inner_dim == 3,
|
/external/tensorflow/tensorflow/core/kernels/ |
H A D | cross_op.cc | 60 auto inner_dim = in0.dim_size(in0.dims() - 1); variable 61 OP_REQUIRES(context, inner_dim == 3,
|
H A D | training_ops.cc | 938 int64 inner_dim = 1; variable 943 inner_dim *= grad.dim_size(d); 950 OP_REQUIRES(ctx, inner_dim > 0, 955 if (inner_dim > 1) { 1278 int64 inner_dim = 1; variable 1283 inner_dim *= grad.dim_size(d); 1291 OP_REQUIRES(ctx, inner_dim > 0, 1296 if (inner_dim > 1) { 1429 int64 inner_dim = 1; variable 1434 inner_dim * 1718 int64 inner_dim = 1; variable 2070 int64 inner_dim = 1; variable [all...] |
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
H A D | TensorConvolution.h | 875 const int inner_dim = ((maxSharedMem / (sizeof(Scalar)) - kernel_size + 1 + 31) / 32) * 32; local 876 maxX = numext::mini<int>(inner_dim, numX); 883 const int inner_dim = maxSharedMem / ((warpSize + kernel_size) * sizeof(Scalar)); local 884 const int maxP = numext::mini<int>(inner_dim, numP); 885 maxX = numext::mini<int>(maxSharedMem / (inner_dim * sizeof(Scalar)) - kernel_size + 1, numX); 938 int inner_dim = ((static_cast<int>(scaling_factor * kernel_size_x) - kernel_size_x + 1 + 32) / 32) * 32; local 939 const int maxX = numext::mini<int>(inner_dim, numX);
|
Completed in 223 milliseconds