Searched refs:input_dims (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/runtime/test/specs/V1_3/ |
| D | fill.mod.py | 16 def test(name, input_dims, value, output, input_dims_data, output_data): argument
17 model = Model().Operation("FILL", input_dims, value).To(output)
19 input_dims: input_dims_data,
28 input_dims=Input("input0", "TENSOR_INT32", "{1}"),
37 input_dims=Input("input0", "TENSOR_INT32", "{3}"),
46 input_dims=Input("input0", "TENSOR_INT32", "{5}"),
|
| /frameworks/ml/nn/common/operations/ |
| D | QuantizedLSTM.cpp | 58 void quantizedLstmStep(const uint8_t* input_data_uint8, const Dims<4>& input_dims, in quantizedLstmStep() argument
71 MatchingFlatSizeSkipDim(input_dims, 0, prev_activ_dims, prevCellState_dims, in quantizedLstmStep()
75 const int input_depth = ArraySize(input_dims, 0); in quantizedLstmStep()
97 Dims<4> const* concat_input_arrays_dims[2] = {&input_dims, &prev_activ_dims}; in quantizedLstmStep()
|