Home
last modified time | relevance | path

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

/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DQLSTMLayerNormalization.cpp57 const uint32_t num_batch = src.shape()[1]; in qlstm_layer_normalization() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
H A Dtrt_engine_utils.cc119 int num_batch, in SetTrtEngineInputs()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
H A Dtrt_engine_op.cc983 const int num_batch = in ExecuteTrtEngine() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_tfl.cc1501 int64_t num_batch = input_type.getNumElements() / num_elems; in matchAndRewrite() local