Searched defs:bat (Results 1 - 14 of 14) sorted by relevance

/frameworks/ml/nn/runtime/test/specs/V1_0/
H A Davg_pool_float_2.mod.py20 bat = 5 variable
25 i0 = Input("i0", "TENSOR_FLOAT32", "{%d, %d, %d, %d}" % (bat, row, col, chn))
39 "{%d, %d, %d, %d}" % (bat, output_row, output_col, chn))
45 input_values = [1. for _ in range(bat * row * col * chn)]
47 output_values = [1. for _ in range(bat * output_row * output_col * chn)]
H A Davg_pool_float_3.mod.py20 bat = 1 variable
25 i0 = Input("i0", "TENSOR_FLOAT32", "{%d, %d, %d, %d}" % (bat, row, col, chn))
39 "{%d, %d, %d, %d}" % (bat, output_row, output_col, chn))
45 input_values = [x % 2 for x in range(bat * row * col * chn)]
47 output_values = [.5 for _ in range(bat * output_row * output_col * chn)]
H A Davg_pool_float_4.mod.py20 bat = 5 variable
25 i0 = Input("i0", "TENSOR_FLOAT32", "{%d, %d, %d, %d}" % (bat, row, col, chn))
39 "{%d, %d, %d, %d}" % (bat, output_row, output_col, chn))
45 input_values = [10 for _ in range(bat * row * col * chn)]
47 output_values = [6 for _ in range(bat * output_row * output_col * chn)]
H A Davg_pool_quant8_2.mod.py20 bat = 5 variable
25 i0 = Input("i0", "TENSOR_QUANT8_ASYMM", "{%d, %d, %d, %d}, 0.5f, 0" % (bat, row, col, chn))
39 "{%d, %d, %d, %d}, 0.5f, 0" % (bat, output_row, output_col, chn))
45 input_values = [255 for _ in range(bat * row * col * chn)]
47 output_values = [255 for _ in range(bat * output_row * output_col * chn)]
H A Davg_pool_quant8_3.mod.py20 bat = 1 variable
25 i0 = Input("i0", "TENSOR_QUANT8_ASYMM", "{%d, %d, %d, %d}, 0.5f, 0" % (bat, row, col, chn))
39 "{%d, %d, %d, %d}, 0.5f, 0" % (bat, output_row, output_col, chn))
45 input_values = [x % 4 * 2 for x in range(bat * row * col * chn)]
47 output_values = [3 for _ in range(bat * output_row * output_col * chn)]
H A Dmax_pool_float_2.mod.py20 bat = 5 variable
25 i0 = Input("i0", "TENSOR_FLOAT32", "{%d, %d, %d, %d}" % (bat, row, col, chn))
39 "{%d, %d, %d, %d}" % (bat, output_row, output_col, chn))
45 input_range = bat * row * col * chn
48 output_range = bat * output_row * output_col * chn
H A Dmax_pool_float_3.mod.py20 bat = 5 variable
25 i0 = Input("i0", "TENSOR_FLOAT32", "{%d, %d, %d, %d}" % (bat, row, col, chn))
39 "{%d, %d, %d, %d}" % (bat, output_row, output_col, chn))
45 input_range = bat * row * col * chn
48 output_range = bat * output_row * output_col * chn
H A Dmax_pool_quant8_2.mod.py20 bat = 5 variable
25 i0 = Input("i0", "TENSOR_QUANT8_ASYMM", "{%d, %d, %d, %d}, 0.5f, 0" % (bat, row, col, chn))
39 "{%d, %d, %d, %d}, 0.5f, 0" % (bat, output_row, output_col, chn))
45 input_range = bat * row * col * chn
48 output_range = bat * output_row * output_col * chn
H A Dmax_pool_quant8_3.mod.py20 bat = 5 variable
25 i0 = Input("i0", "TENSOR_QUANT8_ASYMM", "{%d, %d, %d, %d}, 0.5f, 0" % (bat, row, col, chn))
39 "{%d, %d, %d, %d}, 0.5f, 0" % (bat, output_row, output_col, chn))
45 input_range = bat * row * col * chn
48 output_range = bat * output_row * output_col * chn
/frameworks/ml/nn/runtime/test/specs/V1_1/
H A Davg_pool_float_2_relaxed.mod.py20 bat = 5 variable
25 i0 = Input("i0", "TENSOR_FLOAT32", "{%d, %d, %d, %d}" % (bat, row, col, chn))
39 "{%d, %d, %d, %d}" % (bat, output_row, output_col, chn))
46 input_values = [1. for _ in range(bat * row * col * chn)]
48 output_values = [1. for _ in range(bat * output_row * output_col * chn)]
H A Davg_pool_float_3_relaxed.mod.py20 bat = 1 variable
25 i0 = Input("i0", "TENSOR_FLOAT32", "{%d, %d, %d, %d}" % (bat, row, col, chn))
39 "{%d, %d, %d, %d}" % (bat, output_row, output_col, chn))
46 input_values = [x % 2 for x in range(bat * row * col * chn)]
48 output_values = [.5 for _ in range(bat * output_row * output_col * chn)]
H A Davg_pool_float_4_relaxed.mod.py20 bat = 5 variable
25 i0 = Input("i0", "TENSOR_FLOAT32", "{%d, %d, %d, %d}" % (bat, row, col, chn))
39 "{%d, %d, %d, %d}" % (bat, output_row, output_col, chn))
46 input_values = [10 for _ in range(bat * row * col * chn)]
48 output_values = [6 for _ in range(bat * output_row * output_col * chn)]
H A Dmax_pool_float_2_relaxed.mod.py20 bat = 5 variable
25 i0 = Input("i0", "TENSOR_FLOAT32", "{%d, %d, %d, %d}" % (bat, row, col, chn))
39 "{%d, %d, %d, %d}" % (bat, output_row, output_col, chn))
46 input_range = bat * row * col * chn
49 output_range = bat * output_row * output_col * chn
H A Dmax_pool_float_3_relaxed.mod.py20 bat = 5 variable
25 i0 = Input("i0", "TENSOR_FLOAT32", "{%d, %d, %d, %d}" % (bat, row, col, chn))
39 "{%d, %d, %d, %d}" % (bat, output_row, output_col, chn))
46 input_range = bat * row * col * chn
49 output_range = bat * output_row * output_col * chn

Completed in 478 milliseconds