Home
last modified time | relevance | path

Searched defs:kTensorSize (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/federated-compute/fcp/aggregation/tensorflow/
H A Dtensorflow_checkpoint_builder_factory_test.cc98 static constexpr int kTensorSize = 1024; in TEST() local
/aosp_15_r20/external/pytorch/binaries/
H A Drecord_function_benchmark.cc16 const int kTensorSize = 16; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_level_cost_estimator_test.cc1093 const int kTensorSize = 1000; in TEST_F() local
2095 const int kTensorSize = 1000; in TEST_F() local
2126 const int kTensorSize = 1000; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dinterpreter_test.cc97 constexpr int kTensorSize = 16; in TEST() local
115 constexpr int kTensorSize = 16; in TEST() local