Home
last modified time | relevance | path

Searched defs:set_batch_size (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dweights.h253 void set_batch_size(int batch_size) { batch_size_ = batch_size; } in set_batch_size() function
/aosp_15_r20/out/soong/.intermediates/external/federated-compute/federated-compute-cc-proto-lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/fcp/protos/
Dplan.pb.h15345 inline void DatasetInput::set_batch_size(::int32_t value) { in set_batch_size() function
15499 void DatasetInputPlaceholders::set_batch_size(ArgT0&& arg0, ArgT... args) { in set_batch_size() function