Home
last modified time | relevance | path

Searched defs:kBatch (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dselect_v2_test_util.cc142 const int kBatch = 1; in SelectV2TestTemplate() local
171 const int kBatch = 4; in SelectV2BatchTestTemplate() local
199 const int kBatch = 1; in SelectV2BroadcastFalseTestTemplate() local
229 const int kBatch = 1; in SelectV2BroadcastTrueTestTemplate() local
259 const int kBatch = 1; in SelectV2BroadcastBothTestTemplate() local
289 const int kBatch = 1; in SelectV2ChannelsTestTemplate() local
318 const int kBatch = 3; in SelectV2ChannelsBatchTestTemplate() local
348 const int kBatch = 1; in SelectV2ChannelsBroadcastFalseTestTemplate() local
/external/tensorflow/tensorflow/core/util/
Deinsum_op_util.h40 kBatch = 1, enumerator
/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.h42 kBatch = 0, enumerator
Dmemory_space_assignment_test.cc6057 constexpr int kBatch = 8; in TEST_P() local
6095 constexpr int kBatch = 8; in TEST_P() local
6137 constexpr int kBatch = 8; in TEST_P() local
6179 constexpr int kBatch = 8; in TEST_P() local
6225 constexpr int kBatch = 8; in TEST_P() local
6290 constexpr int kBatch = 8; in TEST_P() local
6324 constexpr int kBatch = 8; in TEST_P() local
6362 constexpr int kBatch = 2; in TEST_P() local
6410 constexpr int kBatch = 2; in TEST_P() local
6463 constexpr int kBatch = 8; in TEST_P() local
[all …]
/external/cronet/stable/base/win/
Dsid.h39 kBatch, enumerator
/external/cronet/tot/base/win/
Dsid.h39 kBatch, enumerator
/external/skia/bench/
DBulkRectBench.cpp30 kBatch, // Bulk API submission, one call to draw every rectangle enumerator
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtensor_utils_test.cc349 constexpr int kBatch = 2; in TEST() local
1479 const int kBatch = 2; in TEST() local
1562 const int kBatch = 2; in TEST() local
1915 constexpr int kBatch = 2; in TEST() local
1928 constexpr int kBatch = 2; in TEST() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
Deinsum.cc69 constexpr auto kBatch = DimType::kBatch; variable
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddynamic_ops_test.cc514 const int32_t kBatch = operand_shape[1]; in RunR3Contiguous() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_ops_test.cc922 void VerifyFusedMatMul(const int kBatch, const int kInputChannel, in VerifyFusedMatMul()