Searched refs:kMaxDims (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | broadcast_to.cc | 34 constexpr int kMaxDims = 8; variable
57 TF_LITE_ENSURE_MSG(context, output_num_dims <= kMaxDims, in ResizeOutputTensor()
93 (NumDimensions(GetInput(context, node, 0)) <= kMaxDims), in Prepare()
119 reference_ops::BroadcastTo<kMaxDims>( in Eval()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | mirror_pad_op.cc | 68 constexpr int kMaxDims = 5; in Compute() local
69 OP_REQUIRES(context, kMinDims <= dims && dims <= kMaxDims, in Compute()
71 kMaxDims, "]: ", dims)); in Compute()
279 constexpr int kMaxDims = 5; in Compute() local
280 OP_REQUIRES(context, kMinDims <= dims && dims <= kMaxDims, in Compute()
282 kMaxDims, "]: ", dims)); in Compute()
|
| /external/angle/src/tests/gl_tests/ |
| D | UniformTest.cpp | 1301 constexpr size_t kMaxDims = 4; in TEST_P() local
1303 GLfloat matrixValues[kMaxDims * kMaxDims]; in TEST_P()
1305 for (size_t i = 0; i < kMaxDims * kMaxDims; ++i) in TEST_P()
1311 UniformMatrixCxRfv uniformMatrixCxRfv[kMaxDims + 1][kMaxDims + 1] = { in TEST_P()
1321 for (size_t cols = kMinDims; cols <= kMaxDims; ++cols) in TEST_P()
1323 for (size_t rows = kMinDims; rows <= kMaxDims; ++rows) in TEST_P()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pad_op.cc | 52 static const int kMaxDims = 8; in Compute() local
53 OP_REQUIRES(context, kMinDims <= dims && dims <= kMaxDims, in Compute()
55 kMaxDims, "]: ", dims)); in Compute()
|