| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | scatter_op_test.cc | 283 void BM_ScatterHelper(::testing::benchmark::State& state, int embedding_size, in BM_ScatterHelper()
317 const int embedding_size = state.range(0); in BM_ScatterUpdateInt32() local
322 const int embedding_size = state.range(0); in BM_ScatterUpdateInt64() local
328 const int embedding_size = state.range(0); in BM_ScatterAddInt32() local
334 const int embedding_size = state.range(0); in BM_ScatterAddInt32Large() local
339 const int embedding_size = state.range(0); in BM_ScatterAddInt64() local
345 const int embedding_size = state.range(0); in BM_ScatterMulInt32() local
350 const int embedding_size = state.range(0); in BM_ScatterMulInt64() local
356 const int embedding_size = state.range(0); in BM_ScatterDivInt32() local
361 const int embedding_size = state.range(0); in BM_ScatterDivInt64() local
[all …]
|
| H A D | scatter_nd_op_test.cc | 257 void BM_ScatterNdHelper(::testing::benchmark::State& state, int embedding_size, in BM_ScatterNdHelper()
291 const int embedding_size = state.range(0); in BM_ScatterNdUpdateInt32() local
296 const int embedding_size = state.range(0); in BM_ScatterNdUpdateInt64() local
302 const int embedding_size = state.range(0); in BM_ScatterNdAddInt32() local
307 const int embedding_size = state.range(0); in BM_ScatterNdAddInt64() local
|
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_static_module.cc | 486 const int embedding_size = 32; in TEST() local
514 const int embedding_size = 32; in TEST() local
555 const int embedding_size = 32; in TEST() local
616 const int embedding_size = 32; in TEST() local
744 const int embedding_size = 32; in TEST() local
768 const int embedding_size = 32; in TEST() local
880 const int embedding_size = 32; in TEST() local
|
| H A D | deep_wide_pt_bench.cc | 5 const int embedding_size = 32; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | embedding_lookup_sparse.cc | 127 float current_squares_weight, int embedding_size, in FinalizeAggregation()
180 size_t embedding_size = 1; in Eval() local
|
| /aosp_15_r20/external/libtextclassifier/native/actions/ |
| H A D | feature-processor.cc | 86 const int embedding_size = options_->embedding_size(); in AppendFeatures() local
|
| /aosp_15_r20/external/libtextclassifier/native/annotator/ |
| H A D | model-executor.cc | 46 const flatbuffers::Vector<uint8_t>* model_spec_buffer, int embedding_size, in FromBuffer()
|
| H A D | feature-processor.cc | 867 const int embedding_size = GetOptions()->embedding_size(); in AppendTokenFeaturesWithCache() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/ |
| H A D | NestedTensorBinaryOps.cu | 82 const auto embedding_size = other.size(2); in _nested_op_dense_esuhm_kernel() local
|
| /aosp_15_r20/external/cronet/third_party/icu/source/common/ |
| H A D | lstmbe.cpp | 392 int32_t embedding_size = ures_getInt(embeddings_res.getAlias(), &status); in LSTMData() local
|
| /aosp_15_r20/external/icu/icu4c/source/common/ |
| H A D | lstmbe.cpp | 392 int32_t embedding_size = ures_getInt(embeddings_res.getAlias(), &status); in LSTMData() local
|
| /aosp_15_r20/out/soong/.intermediates/external/libtextclassifier/native/libtextclassifier_fbgen_actions_actions_model/gen/actions/ |
| D | actions_model_generated.h | 808 int32_t embedding_size = -1; member
861 int32_t embedding_size() const { in embedding_size() function
|
| /aosp_15_r20/out/soong/.intermediates/external/libtextclassifier/native/libtextclassifier_fbgen_annotator_model/gen/annotator/ |
| D | model_generated.h | 3870 int32_t embedding_size = -1; member
3947 int32_t embedding_size() const { in embedding_size() function
|