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()
308 int32_t SingleOpModel::GetTensorSize(int index) const { in GetTensorSize() function in tflite::SingleOpModel
630 tensor_size = GetTensorSize(index); in ExtractVector()671 int32_t GetTensorSize(int index) const;
877 std::vector<float> zero(GetTensorSize(input_), 0.0); in Init()