Searched refs:kElementCount (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stateful_random_ops.h | 35 (PhiloxRandom::ResultType::kElementCount +
36 PhiloxRandom::Key::kElementCount) /
|
| /external/angle/src/tests/gl_tests/ |
| D | ShaderStorageBufferTest.cpp | 282 constexpr unsigned int kElementCount = 2; in TEST_P() local
288 glBufferData(GL_SHADER_STORAGE_BUFFER, kElementCount * kArrayStride, nullptr, GL_STATIC_DRAW); in TEST_P()
301 void *ptr = glMapBufferRange(GL_SHADER_STORAGE_BUFFER, 0, kElementCount * kArrayStride, in TEST_P()
303 for (unsigned int idx = 0; idx < kElementCount; idx++) in TEST_P()
433 constexpr unsigned int kElementCount = 2; in TEST_P() local
443 constexpr int kBufferSize = kElementCount * kArrayStride; in TEST_P()
470 for (unsigned int idx = 0; idx < kElementCount; idx++) in TEST_P()
479 for (unsigned int idx = 0; idx < kElementCount; idx++) in TEST_P()
1153 constexpr unsigned int kElementCount = 2; in TEST_P() local
1157 constexpr unsigned int kExpectedValues[kElementCount][kComponentCount] = {{1u, 2u}, {3u, 4u}}; in TEST_P()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_execute_test.cc | 411 const int kElementCount = 100; in XLA_TEST_F() local
415 std::vector<Shape> element_shapes(kElementCount, element_shape); in XLA_TEST_F()
423 for (int i = 0; i < kElementCount; ++i) { in XLA_TEST_F()
433 for (int i = 0; i < kElementCount; ++i) { in XLA_TEST_F()
442 for (int i = 0; i < kElementCount; ++i) { in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/core/lib/random/ |
| D | philox_random.h | 52 static constexpr int kElementCount = ElementCount;
|