Searched refs:GetTensorSize (Results 1 – 7 of 7) sorted by relevance
123 std::vector<int16_t> initial_state(GetTensorSize(cell_state_out_), 0); in QuantizedLSTMOpModel()125 std::vector<uint8_t> initial_prev_output(GetTensorSize(output_), 0); in QuantizedLSTMOpModel()136 std::vector<uint8_t> zero(GetTensorSize(zero_input_), 128); in setInput()
48 int32_t GetOutputSize() { return GetTensorSize(output_); } in GetOutputSize()
59 int32_t GetOutputSize() { return GetTensorSize(output_); } in GetOutputSize()
52 int32_t GetOutputSize() { return GetTensorSize(output_); } in GetOutputSize()
274 int32_t SingleOpModel::GetTensorSize(int index) const { in GetTensorSize() function in tflite::SingleOpModel
586 tensor_size = GetTensorSize(index); in ExtractVector()619 int32_t GetTensorSize(int index) const;
883 std::vector<float> zero(GetTensorSize(input_), 0.0); in Init()