Searched refs:TF_ARRAYSIZE (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/core/lib/random/ |
| D | distribution_sampler_test.cc | 79 TestDistribution(kEven2, TF_ARRAYSIZE(kEven2)); in TEST_F()
80 TestDistribution(kEven3, TF_ARRAYSIZE(kEven3)); in TEST_F()
81 TestDistribution(kEven4, TF_ARRAYSIZE(kEven4)); in TEST_F()
82 TestDistribution(kDist1, TF_ARRAYSIZE(kDist1)); in TEST_F()
|
| D | weighted_picker_test.cc | 104 TestPickAt(TF_ARRAYSIZE(weights), weights); in TEST()
|
| /external/tensorflow/tensorflow/core/lib/gtl/ |
| D | iterator_range_test.cc | 56 ASSERT_LT(index, TF_ARRAYSIZE(v)); in TEST()
68 ASSERT_LT(index, TF_ARRAYSIZE(v)); in TEST()
|
| /external/tensorflow/tensorflow/core/lib/core/ |
| D | arena_test.cc | 33 for (size_t i = 0; i < TF_ARRAYSIZE(tmp); i++) { in TestMemory()
40 for (size_t i = 0; i < TF_ARRAYSIZE(tmp); i++) { in TestMemory()
|
| D | arena.cc | 139 if (blocks_alloced_ < TF_ARRAYSIZE(first_blocks_)) { in AllocNewBlock()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | substr_op_test.cc | 116 const int sz = TF_ARRAYSIZE(ascii_lines); in GetTestTensor()
126 const int sz = TF_ARRAYSIZE(unicode_lines); in GetTestUTF8Tensor()
|
| D | string_split_op_test.cc | 58 const int sz = TF_ARRAYSIZE(lines); in GetTestTensor()
|
| D | regex_replace_op_test.cc | 61 const int sz = TF_ARRAYSIZE(lines); in GetTestTensor()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 1093 TF_SessionPRunSetup(sess, feeds, TF_ARRAYSIZE(feeds), fetches, in TEST()
1094 TF_ARRAYSIZE(fetches), nullptr, 0, &handle, s); in TEST()
1140 Int8Tensor(vec2_dims, TF_ARRAYSIZE(vec2_dims), data), TF_DeleteTensor); in TEST()
1146 Int8Tensor(vec3_dims, TF_ARRAYSIZE(vec3_dims), data), TF_DeleteTensor); in TEST()
1210 TF_AddInputList(desc_, inputs, TF_ARRAYSIZE(inputs)); in SetUp()
2033 const size_t list_size = TF_ARRAYSIZE(list); in TEST_F()
2064 const size_t list_size = TF_ARRAYSIZE(list); in TEST_F()
2095 const size_t list_size = TF_ARRAYSIZE(list); in TEST_F()
2126 const size_t list_size = TF_ARRAYSIZE(list); in TEST_F()
2153 const size_t sz = TF_ARRAYSIZE(partial_shape); in TEST_F()
[all …]
|
| D | c_api_function_test.cc | 1693 TF_AddInputList(desc, inputs, TF_ARRAYSIZE(inputs)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | parse_flags_from_env_test.cc | 89 for (int i = 0; i != TF_ARRAYSIZE(kTempVars) && tmp_dir == nullptr; i++) { in TEST()
127 for (int i = 0; i != TF_ARRAYSIZE(test); i++) { in TEST()
|
| D | literal_test.cc | 993 std::vector<int64> source_indexes(TF_ARRAYSIZE(dimensions), 0); in TEST_F()
994 std::vector<int64> blank_indexes(TF_ARRAYSIZE(dimensions), 0); in TEST_F()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | colors.cc | 38 return kColors[dindex % TF_ARRAYSIZE(kColors)]; in ColorFor()
|
| /external/tensorflow/tensorflow/contrib/ffmpeg/ |
| D | decode_audio_op.cc | 146 kValidFileFormats, kValidFileFormats + TF_ARRAYSIZE(kValidFileFormats)); in Compute()
225 kValidFileFormats, kValidFileFormats + TF_ARRAYSIZE(kValidFileFormats)); in DecodeAudioOp()
|
| /external/tensorflow/tensorflow/core/platform/ |
| D | macros.h | 104 #define TF_ARRAYSIZE(a) \ macro
|
| /external/tensorflow/tensorflow/core/lib/strings/ |
| D | numbers.cc | 452 CHECK(unit < units + TF_ARRAYSIZE(units)); in HumanReadableNum()
484 CHECK(unit < units + TF_ARRAYSIZE(units)); in HumanReadableNumBytes()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 612 CHECK(TF_ARRAYSIZE(options) == TF_ARRAYSIZE(option_values)); in LoadPtx()
619 res = cuModuleLoadDataEx(module, ptx_data, TF_ARRAYSIZE(options), options, in LoadPtx()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen_internal.cc | 711 for (size_t i = 0; i < TF_ARRAYSIZE(kAttrTypeName); ++i) { in AddDocStringAttrs()
|