Searched defs:inner_dim (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dcross_op.cc38 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 Dcross_op.cc60 auto inner_dim = in0.dim_size(in0.dims() - 1); variable
61 OP_REQUIRES(context, inner_dim == 3,
H A Dtraining_ops.cc938 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 DTensorConvolution.h875 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