Home
last modified time | relevance | path

Searched defs:total_input_size (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dbatch_matmul_test.cc397 int total_input_size = 1; local
923 int total_input_size = 1; local
H A Dfully_connected_test.cc153 int total_input_size = 1; local
410 int total_input_size = 1; local
1551 int total_input_size = 1; local
H A Dfully_connected.cc480 int total_input_size = 1; in EvalPie() local
518 int total_input_size = 1; in EvalHybridDense() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Dmatmul_test.cc30 int total_input_size = 1; in FullyConnectedOpModel() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dbatch_norm_op.cc116 int total_input_size = ctx->InputShape(0).num_elements(); in CompileImpl() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate_signed_quantization_test.cc75 int total_input_size = 1; local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_level_cost_estimator.cc1576 int64_t total_input_size = 0; in CalculateInputSize() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc2392 int total_input_size = bn_train_input_type_tensor.getNumElements(); in matchAndRewrite() local